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

retries for connecting to broker

Browse Source
This commit is contained in:
Matthew R Kasun
2022-01-27 15:46:45 -05:00
parent 4ccce6bac5
commit 84d9faeb3f
2 changed files with 23 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
mq/mq.go
View File

@@ -156,9 +156,17 @@ func SetupMQTT() mqtt.Client {
opts := mqtt.NewClientOptions() opts := mqtt.NewClientOptions()
broker := servercfg.GetMessageQueueEndpoint() broker := servercfg.GetMessageQueueEndpoint()
opts.AddBroker(broker) opts.AddBroker(broker)
client := mqtt.NewClient(opts) tperiod := time.Now().Add(10 * time.Second)
if token := client.Connect(); token.Wait() && token.Error() != nil { for {
log.Fatal(token.Error()) if token := client.Connect(); token.Wait() && token.Error() != nil {
logger.Log(2, "unable to connect to broker, retrying ...")
if time.Now().Afer(tperiod) {
log.Fatal(0, "could not connect to broker, exiting ...", token.Error())
}
} else {
break
}
time.Sleep(2 * time.Second)
} }
logger.Log(2, "connected to message queue", broker) logger.Log(2, "connected to message queue", broker)
return client return client

14
netclient/functions/daemon.go
View File

@@ -13,6 +13,7 @@ import (
"time" "time"
mqtt "github.com/eclipse/paho.mqtt.golang" mqtt "github.com/eclipse/paho.mqtt.golang"
"github.com/gravitl/netmaker/logger"
"github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/models"
"github.com/gravitl/netmaker/netclient/config" "github.com/gravitl/netmaker/netclient/config"
"github.com/gravitl/netmaker/netclient/local" "github.com/gravitl/netmaker/netclient/local"
@@ -70,8 +71,17 @@ func SetupMQTT(cfg *config.ClientConfig) mqtt.Client {
} }
opts.SetDefaultPublishHandler(All) opts.SetDefaultPublishHandler(All)
client := mqtt.NewClient(opts) client := mqtt.NewClient(opts)
if token := client.Connect(); token.Wait() && token.Error() != nil { tperiod := time.Now().Add(10 * time.Second)
log.Fatal(token.Error()) for {
if token := client.Connect(); token.Wait() && token.Error() != nil {
logger.Log(2, "unable to connect to broker, retrying ...")
if time.Now().Afer(tperiod) {
log.Fatal(0, "could not connect to broker, exiting ...", token.Error())
}
} else {
break
}
time.Sleep(2 * time.Second)
} }
return client return client
} }