mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-06 09:22:42 +08:00
adding force peer push every 5 min
This commit is contained in:
24
mq/mq.go
24
mq/mq.go
@@ -242,6 +242,23 @@ func Keepalive(ctx context.Context) {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-time.After(time.Second * KEEPALIVE_TIMEOUT):
|
||||
sendPeers()
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var counter = make(chan int, 0)
|
||||
|
||||
// sendPeers - retrieve networks, send peer ports to all peers
|
||||
func sendPeers() {
|
||||
var force bool
|
||||
var tmpcount = <-counter + 1
|
||||
counter <- tmpcount
|
||||
if tmpcount == 5 {
|
||||
force = true
|
||||
counter <- 0
|
||||
}
|
||||
networks, err := logic.GetNetworks()
|
||||
if err != nil {
|
||||
logger.Log(1, "error retrieving networks for keepalive", err.Error())
|
||||
@@ -252,7 +269,10 @@ func Keepalive(ctx context.Context) {
|
||||
serverNode.SetLastCheckIn()
|
||||
logic.UpdateNode(&serverNode, &serverNode)
|
||||
if network.DefaultUDPHolePunch == "yes" {
|
||||
if logic.ShouldPublishPeerPorts(&serverNode) {
|
||||
if logic.ShouldPublishPeerPorts(&serverNode) || force {
|
||||
if force {
|
||||
logger.Log(2, "sending scheduled peer update (5 min)")
|
||||
}
|
||||
err = PublishPeerUpdate(&serverNode)
|
||||
if err != nil {
|
||||
logger.Log(1, "error publishing udp port updates for network", network.NetID)
|
||||
@@ -267,8 +287,6 @@ func Keepalive(ctx context.Context) {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// func publishServerKeepalive(client mqtt.Client, network *models.Network) {
|
||||
// nodes, err := logic.GetNetworkNodes(network.NetID)
|
||||
|
@@ -132,7 +132,10 @@ func SetupMQTT(cfg *config.ClientConfig, publish bool) mqtt.Client {
|
||||
ncutils.Log("detected broker connection lost, running pull for " + cfg.Node.Network)
|
||||
_, err := Pull(cfg.Node.Network, true)
|
||||
if err != nil {
|
||||
ncutils.Log("could not run pull, please restart daemon or examine network connectivity --- " + err.Error())
|
||||
ncutils.Log("could not run pull, server unreachable, restarting daemon in 5 minutes..." + err.Error())
|
||||
time.Sleep(time.Minute * 5)
|
||||
ncutils.Log("restarting netclient")
|
||||
daemon.Restart()
|
||||
}
|
||||
ncutils.Log("connection re-established with mqtt server")
|
||||
})
|
||||
|
Reference in New Issue
Block a user