gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-09-10 23:24:32 +08:00
Code Issues Projects Releases Packages Wiki Activity

add mq connection lost handler on server

Browse source
This commit is contained in:
abhishek9686 2024-03-27 14:56:47 +07:00
parent 4cae1b0bb4
commit c82918ad35
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
mq/mq.go
View file

@ -9,6 +9,7 @@ import (
mqtt "github.com/eclipse/paho.mqtt.golang" mqtt "github.com/eclipse/paho.mqtt.golang"
"github.com/gravitl/netmaker/logger" "github.com/gravitl/netmaker/logger"
"github.com/gravitl/netmaker/servercfg" "github.com/gravitl/netmaker/servercfg"
"golang.org/x/exp/slog"
) )
// KEEPALIVE_TIMEOUT - time in seconds for timeout // KEEPALIVE_TIMEOUT - time in seconds for timeout
@ -91,6 +92,13 @@ func SetupMQTT() {
opts.SetOrderMatters(false) opts.SetOrderMatters(false)
opts.SetResumeSubs(true) opts.SetResumeSubs(true)
}) })
opts.SetConnectionLostHandler(func(c mqtt.Client, e error) {
slog.Warn("detected broker connection lost", "err", e.Error())
c.Disconnect(250)
slog.Info("re-initiating MQ connection")
SetupMQTT()
})
mqclient = mqtt.NewClient(opts) mqclient = mqtt.NewClient(opts)
tperiod := time.Now().Add(10 * time.Second) tperiod := time.Now().Add(10 * time.Second)
for { for {