zishuo/netmaker
1
0
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-10-05 16:57:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor server pings

Browse Source
This commit is contained in:
Matthew R Kasun
2022-01-27 09:48:32 -05:00
parent 76ae673363
commit 513f85ede7
5 changed files with 39 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
netclient/functions/daemon.go
View File

@@ -102,11 +102,21 @@ func MessageQueue(ctx context.Context, network string) {
if cfg.DebugOn {
ncutils.Log("subscribed to node updates for node " + cfg.Node.Name + " update/peers/" + cfg.Node.ID)
}
if token := client.Subscribe("serverkeepalive/"+cfg.Node.ID, 0, mqtt.MessageHandler(ServerKeepAlive)); token.Wait() && token.Error() != nil {
log.Fatal(token.Error())
}
if cfg.DebugOn {
ncutils.Log("subscribed to server keepalives")
var id string
for _, server := range cfg.NetworkSettings.DefaultServerAddrs {
if server.IsLeader {
id = server.ID
}
if server.Address != "" {
if token := client.Subscribe("serverkeepalive/"+id, 0, mqtt.MessageHandler(ServerKeepAlive)); token.Wait() && token.Error() != nil {
log.Fatal(token.Error())
}
if cfg.DebugOn {
ncutils.Log("subscribed to server keepalives")
}
} else {
ncutils.Log("leader not defined for network" + cfg.Network)
}
}
defer client.Disconnect(250)
go MonitorKeepalive(ctx, client, &cfg)
@@ -270,6 +280,7 @@ func MonitorKeepalive(ctx context.Context, client mqtt.Client, cfg *config.Clien
func ServerKeepAlive(client mqtt.Client, msg mqtt.Message) {
serverid := string(msg.Payload())
keepalive <- serverid
ncutils.Log("keepalive from server")
}
// Resubscribe --- handles resubscribing if needed