change queue identifier from server.ID to Network.NetID

This commit is contained in:
afeiszli
2022-02-02 11:56:39 -05:00
committed by 0xdcarns
parent f7258bf98f
commit 239b9d36fc
8 changed files with 52 additions and 150 deletions

View File

@@ -107,14 +107,10 @@ func (s *NodeServiceServer) CreateNode(ctx context.Context, req *nodepb.Object)
go func(node *models.Node) { go func(node *models.Node) {
if node.UDPHolePunch == "yes" { if node.UDPHolePunch == "yes" {
var currentServerNodeID, getErr = logic.GetNetworkServerNodeID(node.Network) var currentServerNode, getErr = logic.GetNetworkServerLeader(node.Network)
if getErr != nil { if getErr != nil {
return return
} }
var currentServerNode, currErr = logic.GetNodeByID(currentServerNodeID)
if currErr != nil {
return
}
for i := 0; i < 5; i++ { for i := 0; i < 5; i++ {
if logic.HasPeerConnected(node) { if logic.HasPeerConnected(node) {
if logic.ShouldPublishPeerPorts(&currentServerNode) { if logic.ShouldPublishPeerPorts(&currentServerNode) {
@@ -190,7 +186,6 @@ func getServerAddrs(node *models.Node) {
serverAddrs = append(serverAddrs, models.ServerAddr{ serverAddrs = append(serverAddrs, models.ServerAddr{
IsLeader: logic.IsLeader(&node), IsLeader: logic.IsLeader(&node),
Address: node.Address, Address: node.Address,
ID: node.ID,
}) })
} }

View File

@@ -23,14 +23,10 @@ func runServerPeerUpdate(node *models.Node, ifaceDelta bool) error {
if servercfg.IsClientMode() != "on" { if servercfg.IsClientMode() != "on" {
return nil return nil
} }
var currentServerNodeID, getErr = logic.GetNetworkServerNodeID(node.Network) var currentServerNode, getErr = logic.GetNetworkServerLeader(node.Network)
if err != nil { if err != nil {
return getErr return getErr
} }
var currentServerNode, currErr = logic.GetNodeByID(currentServerNodeID)
if currErr != nil {
return currErr
}
if err = logic.ServerUpdate(&currentServerNode, ifaceDelta); err != nil { if err = logic.ServerUpdate(&currentServerNode, ifaceDelta); err != nil {
logger.Log(1, "server node:", currentServerNode.ID, "failed update") logger.Log(1, "server node:", currentServerNode.ID, "failed update")
return err return err

View File

@@ -5,7 +5,6 @@ import (
"errors" "errors"
"fmt" "fmt"
"sort" "sort"
"strings"
"time" "time"
"github.com/go-playground/validator/v10" "github.com/go-playground/validator/v10"
@@ -237,7 +236,6 @@ func CreateNode(node *models.Node) error {
return fmt.Errorf("invalid address: ipv6 " + node.Address6 + " is not unique") return fmt.Errorf("invalid address: ipv6 " + node.Address6 + " is not unique")
} }
// TODO: This covers legacy nodes, eventually want to remove legacy check
node.ID = uuid.NewString() node.ID = uuid.NewString()
//Create a JWT for the node //Create a JWT for the node
@@ -269,67 +267,6 @@ func CreateNode(node *models.Node) error {
return err return err
} }
// IfaceDelta - is there interface changes
// func IfaceDelta(currentNode *models.Node, newNode *models.Node) bool {
// SetNodeDefaults(newNode)
// // single comparison statements
// if currentNode.IsServer != "yes" {
// return false
// }
// if newNode.Endpoint != currentNode.Endpoint ||
// newNode.LocalAddress != currentNode.LocalAddress ||
// newNode.PublicKey != currentNode.PublicKey ||
// newNode.Address != currentNode.Address ||
// newNode.IsEgressGateway != currentNode.IsEgressGateway ||
// newNode.IsIngressGateway != currentNode.IsIngressGateway ||
// newNode.IsRelay != currentNode.IsRelay ||
// newNode.UDPHolePunch != currentNode.UDPHolePunch ||
// newNode.IsPending != currentNode.IsPending ||
// newNode.PersistentKeepalive != currentNode.PersistentKeepalive ||
// len(newNode.ExcludedAddrs) != len(currentNode.ExcludedAddrs) ||
// len(newNode.AllowedIPs) != len(currentNode.AllowedIPs) {
// return true
// }
// // multi-comparison statements
// if newNode.IsDualStack == "yes" {
// if newNode.Address6 != currentNode.Address6 {
// return true
// }
// }
// if newNode.IsEgressGateway == "yes" {
// if len(currentNode.EgressGatewayRanges) != len(newNode.EgressGatewayRanges) {
// return true
// }
// for _, address := range newNode.EgressGatewayRanges {
// if !StringSliceContains(currentNode.EgressGatewayRanges, address) {
// return true
// }
// }
// }
// if newNode.IsRelay == "yes" {
// if len(currentNode.RelayAddrs) != len(newNode.RelayAddrs) {
// return true
// }
// for _, address := range newNode.RelayAddrs {
// if !StringSliceContains(currentNode.RelayAddrs, address) {
// return true
// }
// }
// }
// for _, address := range newNode.AllowedIPs {
// if !StringSliceContains(currentNode.AllowedIPs, address) {
// return true
// }
// }
// return false
// }
// GetAllNodes - returns all nodes in the DB // GetAllNodes - returns all nodes in the DB
func GetAllNodes() ([]models.Node, error) { func GetAllNodes() ([]models.Node, error) {
var nodes []models.Node var nodes []models.Node
@@ -602,28 +539,35 @@ func GetDeletedNodeByID(uuid string) (models.Node, error) {
} }
// GetNetworkServerNodeID - get network server node ID if exists // GetNetworkServerNodeID - get network server node ID if exists
func GetNetworkServerNodeID(network string) (string, error) { func GetNetworkServerLeader(network string) (models.Node, error) {
var nodes, err = GetNetworkNodes(network) nodes, err := GetSortedNetworkServerNodes(network)
if err != nil { if err != nil {
return "", err return models.Node{}, err
} }
for _, node := range nodes { for _, node := range nodes {
if node.IsServer == "yes" { if IsLeader(&node) {
if servercfg.GetNodeID() != "" { return node, nil
if servercfg.GetNodeID() == node.MacAddress {
if strings.Contains(node.ID, "###") {
DeleteNodeByMacAddress(&node, true)
logger.Log(1, "deleted legacy server node on network "+node.Network)
return "", errors.New("deleted legacy server node on network " + node.Network)
}
return node.ID, nil
}
continue
}
return node.ID, nil
} }
} }
return "", errors.New("could not find server node") return models.Node{}, errors.New("could not find server leader")
}
// GetNetworkServerNodeID - get network server node ID if exists
func GetNetworkServerLocal(network string) (models.Node, error) {
nodes, err := GetSortedNetworkServerNodes(network)
if err != nil {
return models.Node{}, err
}
mac := servercfg.GetNodeID()
if mac == "" {
return models.Node{}, fmt.Errorf("error retrieving local server node: server node ID is unset")
}
for _, node := range nodes {
if mac == node.MacAddress {
return node, nil
}
}
return models.Node{}, errors.New("could not find node for local server")
} }
// validateServer - make sure servers dont change port or address // validateServer - make sure servers dont change port or address

View File

@@ -73,7 +73,7 @@ func GetPeerUpdate(node *models.Node) (models.PeerUpdate, error) {
} }
peers = append(peers, peerData) peers = append(peers, peerData)
if peer.IsServer == "yes" { if peer.IsServer == "yes" {
serverNodeAddresses = append(serverNodeAddresses, models.ServerAddr{ID: peer.ID, IsLeader: IsLeader(&peer), Address: peer.Address}) serverNodeAddresses = append(serverNodeAddresses, models.ServerAddr{IsLeader: IsLeader(&peer), Address: peer.Address})
} }
} }
if node.IsIngressGateway == "yes" { if node.IsIngressGateway == "yes" {

View File

@@ -177,7 +177,6 @@ type Telemetry struct {
// ServerAddr - to pass to clients to tell server addresses and if it's the leader or not // ServerAddr - to pass to clients to tell server addresses and if it's the leader or not
type ServerAddr struct { type ServerAddr struct {
ID string `json:"id" bson:"id" yaml:"id"`
IsLeader bool `json:"isleader" bson:"isleader" yaml:"isleader"` IsLeader bool `json:"isleader" bson:"isleader" yaml:"isleader"`
Address string `json:"address" bson:"address" yaml:"address"` Address string `json:"address" bson:"address" yaml:"address"`
} }

View File

@@ -195,13 +195,7 @@ func Keepalive(ctx context.Context) {
logger.Log(1, "error retrieving networks for keepalive", err.Error()) logger.Log(1, "error retrieving networks for keepalive", err.Error())
} }
for _, network := range networks { for _, network := range networks {
var id string serverNode, errN := logic.GetNetworkServerLeader(network.NetID)
for _, servAddr := range network.DefaultServerAddrs {
if servAddr.IsLeader {
id = servAddr.ID
}
}
serverNode, errN := logic.GetNodeByID(id)
if errN == nil { if errN == nil {
serverNode.SetLastCheckIn() serverNode.SetLastCheckIn()
logic.UpdateNode(&serverNode, &serverNode) logic.UpdateNode(&serverNode, &serverNode)
@@ -210,14 +204,19 @@ func Keepalive(ctx context.Context) {
} }
err = PublishPeerUpdate(&serverNode) err = PublishPeerUpdate(&serverNode)
if err != nil { if err != nil {
logger.Log(1, "error publishing udp port updates", err.Error()) logger.Log(1, "error publishing udp port updates for network", network.NetID)
logger.Log(1, errN.Error())
} }
} } else {
if id == "" { logger.Log(1, "unable to retrieve leader for network ", network.NetID)
logger.Log(0, "leader not defined for network", network.NetID) logger.Log(1, errN.Error())
continue continue
} }
if token := client.Publish("serverkeepalive/"+id, 0, false, servercfg.GetVersion()); token.Wait() && token.Error() != nil { if serverNode.Address == "" {
logger.Log(1, "leader not defined for network ", network.NetID)
continue
}
if token := client.Publish("serverkeepalive/"+network.NetID, 0, false, servercfg.GetVersion()); token.Wait() && token.Error() != nil {
logger.Log(1, "error publishing server keepalive for network", network.NetID, token.Error().Error()) logger.Log(1, "error publishing server keepalive for network", network.NetID, token.Error().Error())
} else { } else {
logger.Log(2, "keepalive sent for network", network.NetID) logger.Log(2, "keepalive sent for network", network.NetID)

View File

@@ -3,7 +3,6 @@ package functions
import ( import (
"context" "context"
"encoding/json" "encoding/json"
"errors"
"fmt" "fmt"
"os" "os"
"os/signal" "os/signal"
@@ -161,25 +160,24 @@ func MessageQueue(ctx context.Context, network string) {
if cfg.DebugOn { if cfg.DebugOn {
ncutils.Log(fmt.Sprintf("subscribed to peer updates for node %s peers/%s/%s", cfg.Node.Name, cfg.Node.Network, cfg.Node.ID)) ncutils.Log(fmt.Sprintf("subscribed to peer updates for node %s peers/%s/%s", cfg.Node.Name, cfg.Node.Network, cfg.Node.ID))
} }
var id string
var found bool var found bool
for _, server := range cfg.NetworkSettings.DefaultServerAddrs { for _, server := range cfg.NetworkSettings.DefaultServerAddrs {
if server.IsLeader { if !server.IsLeader {
id = server.ID continue
} }
if server.Address != "" { if server.Address != "" {
if token := client.Subscribe("serverkeepalive/"+id, 0, mqtt.MessageHandler(ServerKeepAlive)); token.Wait() && token.Error() != nil { if token := client.Subscribe("serverkeepalive/"+cfg.Node.Network, 0, mqtt.MessageHandler(ServerKeepAlive)); token.Wait() && token.Error() != nil {
ncutils.Log(token.Error().Error()) ncutils.Log(token.Error().Error())
return return
} }
found = true found = true
if cfg.DebugOn { if cfg.DebugOn {
ncutils.Log("subscribed to server keepalives for server " + id) ncutils.Log("subscribed to server keepalives for server " + cfg.Node.Network)
} }
} }
} }
if !found { if !found {
ncutils.Log("leader not defined for network " + cfg.Network) ncutils.Log("leader not defined for network " + cfg.Node.Network)
} }
defer client.Disconnect(250) defer client.Disconnect(250)
go MonitorKeepalive(ctx, client, &cfg) go MonitorKeepalive(ctx, client, &cfg)
@@ -342,19 +340,13 @@ func UpdatePeers(client mqtt.Client, msg mqtt.Message) {
// MonitorKeepalive - checks time last server keepalive received. If more than 3+ minutes, notify and resubscribe // MonitorKeepalive - checks time last server keepalive received. If more than 3+ minutes, notify and resubscribe
func MonitorKeepalive(ctx context.Context, client mqtt.Client, cfg *config.ClientConfig) { func MonitorKeepalive(ctx context.Context, client mqtt.Client, cfg *config.ClientConfig) {
var id string
for _, servAddr := range cfg.NetworkSettings.DefaultServerAddrs {
if servAddr.IsLeader {
id = servAddr.ID
}
}
for { for {
select { select {
case <-ctx.Done(): case <-ctx.Done():
return return
case <-time.After(time.Second * 150): case <-time.After(time.Second * 150):
var keepalivetime time.Time var keepalivetime time.Time
keepaliveval, ok := keepalive.Load(id) keepaliveval, ok := keepalive.Load(cfg.Node.Network)
if ok { if ok {
keepalivetime = keepaliveval.(time.Time) keepalivetime = keepaliveval.(time.Time)
} else { } else {
@@ -374,15 +366,7 @@ func MonitorKeepalive(ctx context.Context, client mqtt.Client, cfg *config.Clien
// ServerKeepAlive -- handler to react to keepalive messages published by server // ServerKeepAlive -- handler to react to keepalive messages published by server
func ServerKeepAlive(client mqtt.Client, msg mqtt.Message) { func ServerKeepAlive(client mqtt.Client, msg mqtt.Message) {
// var mu sync.Mutex keepalive.Store(parseNetworkFromTopic(msg.Topic()), time.Now())
// mu.Lock()
// defer mu.Unlock()
serverid, err := getID(msg.Topic())
if err != nil {
ncutils.Log("invalid ID in serverkeepalive topic")
}
keepalive.Store(serverid, time.Now())
// ncutils.Log("keepalive from server")
} }
// Resubscribe --- handles resubscribing if needed // Resubscribe --- handles resubscribing if needed
@@ -402,25 +386,24 @@ func Resubscribe(client mqtt.Client, cfg *config.ClientConfig) error {
ncutils.Log("error resubscribing to peers for " + cfg.Node.Network) ncutils.Log("error resubscribing to peers for " + cfg.Node.Network)
return token.Error() return token.Error()
} }
var id string
var found bool var found bool
for _, server := range cfg.NetworkSettings.DefaultServerAddrs { for _, server := range cfg.NetworkSettings.DefaultServerAddrs {
if server.IsLeader { if !server.IsLeader {
id = server.ID continue
} }
if server.Address != "" { if server.Address != "" {
if token := client.Subscribe("serverkeepalive/"+id, 0, mqtt.MessageHandler(ServerKeepAlive)); token.Wait() && token.Error() != nil { if token := client.Subscribe("serverkeepalive/"+cfg.Node.Network, 0, mqtt.MessageHandler(ServerKeepAlive)); token.Wait() && token.Error() != nil {
ncutils.Log("error resubscribing to serverkeepalive for " + cfg.Node.Network) ncutils.Log("error resubscribing to serverkeepalive for " + cfg.Node.Network)
return token.Error() return token.Error()
} }
found = true found = true
if cfg.DebugOn { if cfg.DebugOn {
ncutils.Log("subscribed to server keepalives for server " + id) ncutils.Log("subscribed to server keepalives for server " + cfg.Node.Network)
} }
} }
} }
if !found { if !found {
ncutils.Log("leader not defined for network " + cfg.Network) ncutils.Log("leader not defined for network " + cfg.Node.Network)
} }
ncutils.Log("finished re subbing") ncutils.Log("finished re subbing")
return nil return nil
@@ -583,13 +566,3 @@ func shouldResub(currentServers, newServers []models.ServerAddr) bool {
} }
return false return false
} }
func getID(topic string) (string, error) {
parts := strings.Split(topic, "/")
count := len(parts)
if count == 1 {
return "", errors.New("invalid topic")
}
//the last part of the topic will be the network.ID
return parts[count-1], nil
}

View File

@@ -44,12 +44,8 @@ func SyncServerNetwork(network string) error {
} }
} }
serverNodeID, err := logic.GetNetworkServerNodeID(network) serverNode, err := logic.GetNetworkServerLocal(network)
if !ifaceExists && (err == nil && serverNodeID != "") { if !ifaceExists && (err == nil && serverNode.ID != "") {
serverNode, err := logic.GetNodeByID(serverNodeID)
if err != nil {
return err
}
return logic.ServerUpdate(&serverNode, true) return logic.ServerUpdate(&serverNode, true)
} else if !ifaceExists { } else if !ifaceExists {
_, err := logic.ServerJoin(&serverNetworkSettings) _, err := logic.ServerJoin(&serverNetworkSettings)