Ver Fonte

edited MQ detection

0xdcarns há 2 anos atrás
pai
commit
5f3f99fc80
2 ficheiros alterados com 6 adições e 8 exclusões
  1. 5 7
      main.go
  2. 1 1
      servercfg/serverconf.go

+ 5 - 7
main.go

@@ -164,11 +164,7 @@ func startControllers() {
 // Should we be using a context vice a waitgroup????????????
 func runMessageQueue(wg *sync.WaitGroup) {
 	defer wg.Done()
-	brokerHost, secure := servercfg.GetMessageQueueEndpoint()
-	logger.Log(0, "connecting to mq broker at", brokerHost, "with TLS?", fmt.Sprintf("%v", secure))
-	mq.SetupMQTT()
 	ctx, cancel := context.WithCancel(context.Background())
-
 	if servercfg.IsMessageQueueBackend() { // connect to external broker
 		brokerHost, secure := servercfg.GetMessageQueueEndpoint()
 		logger.Log(0, "connecting to mq broker at", brokerHost, "with TLS?", fmt.Sprintf("%v", secure))
@@ -180,9 +176,11 @@ func runMessageQueue(wg *sync.WaitGroup) {
 	go func() {
 		peerUpdate := make(chan *models.Node)
 		go logic.ManageZombies(ctx, peerUpdate)
-		for nodeUpdate := range peerUpdate {
-			if err := mq.NodeUpdate(nodeUpdate); err != nil {
-				logger.Log(0, "failed to send peer update for deleted node: ", nodeUpdate.ID.String(), err.Error())
+		if servercfg.IsMessageQueueBackend() {
+			for nodeUpdate := range peerUpdate {
+				if err := mq.NodeUpdate(nodeUpdate); err != nil {
+					logger.Log(0, "failed to send peer update for deleted node: ", nodeUpdate.ID.String(), err.Error())
+				}
 			}
 		}
 	}()

+ 1 - 1
servercfg/serverconf.go

@@ -338,7 +338,7 @@ func IsMessageQueueBackend() bool {
 		ismessagequeue = false
 	} else if len(config.Config.Server.MessageQueueBackend) == 0 ||
 		config.Config.Server.MessageQueueBackend == "off" ||
-		os.Getenv("MESSAGEQUEUE_BACKEND") == "internal" {
+		config.Config.Server.MessageQueueBackend == "internal" {
 		ismessagequeue = false
 	}
 	return ismessagequeue