gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-02-25 16:44:01 +08:00
Code Issues Projects Releases Packages Wiki Activity

handled sending peer update on client node update

Browse source
This commit is contained in:
0xdcarns 2022-05-12 09:18:33 -04:00
parent 792c04e972
commit 352f80ef4d
1 changed files with 18 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
mq/handlers.go
View file

@ -81,6 +81,7 @@ func UpdateNode(client mqtt.Client, msg mqtt.Message) {
logger.Log(1, "error saving node", err.Error())
return
}
updateNodePeers(&currentNode)
logger.Log(1, "updated node", id, newNode.Name)
}()
}
@ -114,20 +115,25 @@ func ClientPeerUpdate(client mqtt.Client, msg mqtt.Message) {
return
}
case ncutils.DONE:
currentServerNode, err := logic.GetNetworkServerLocal(currentNode.Network)
if err != nil {
return
}
if err := logic.ServerUpdate(&currentServerNode, false); err != nil {
logger.Log(1, "server node:", currentServerNode.ID, "failed update")
return
}
if err := PublishPeerUpdate(&currentNode); err != nil {
logger.Log(1, "error publishing peer update ", err.Error())
return
}
updateNodePeers(&currentNode)
}
logger.Log(1, "sent peer updates after signal received from", id, currentNode.Name)
}()
}
func updateNodePeers(currentNode *models.Node) {
currentServerNode, err := logic.GetNetworkServerLocal(currentNode.Network)
if err != nil {
logger.Log(1, "failed to get server node failed update\n", err.Error())
return
}
if err := logic.ServerUpdate(&currentServerNode, false); err != nil {
logger.Log(1, "server node:", currentServerNode.ID, "failed update")
return
}
if err := PublishPeerUpdate(currentNode); err != nil {
logger.Log(1, "error publishing peer update ", err.Error())
return
}
}