Browse Source

added server name as ID for pub events

0xdcarns 2 years ago
parent
commit
28fa850c1b
1 changed files with 10 additions and 6 deletions
  1. 10 6
      queue/publishers.go

+ 10 - 6
queue/publishers.go

@@ -8,14 +8,14 @@ import (
 	"github.com/gravitl/netmaker/logger"
 	"github.com/gravitl/netmaker/logic"
 	"github.com/gravitl/netmaker/models"
+	"github.com/gravitl/netmaker/servercfg"
 )
 
 // PublishAllPeerUpdate - publishes a peer update to
 // all hosts with current connections
 func PublishAllPeerUpdate() {
-	const publishAllID = "pub-all"
 	event := models.Event{
-		ID:    publishAllID,
+		ID:    servercfg.GetServer(),
 		Topic: models.EventTopics.SendAllHostPeerUpdate,
 	}
 	EventQueue.Enqueue(event)
@@ -47,23 +47,28 @@ func HostUpdate(hostUpdate *models.HostUpdate) error {
 }
 
 func sendNodeUpdate(e *models.Event) {
+	hostID := e.ID
+	e.ID = servercfg.GetServer()
 	data, err := json.Marshal(e)
 	if err != nil {
 		logger.Log(0, "failed to encode node update", err.Error())
 	}
-	if err = publish(data, e.ID); err != nil {
+	if err = publish(data, hostID); err != nil {
 		logger.Log(0, "failed to send node update", err.Error())
 	}
 }
 
 func sendHostUpdate(e *models.Event) {
 	logger.Log(1, "publishing host update to "+e.ID)
+	hostID := e.ID
+	e.ID = servercfg.GetServer()
 	data, err := json.Marshal(e)
 	if err != nil {
 		logger.Log(0, "failed to encode host update", err.Error())
 		return
 	}
-	if err = publish(data, e.ID); err != nil {
+
+	if err = publish(data, hostID); err != nil {
 		logger.Log(0, "failed to send host update", err.Error())
 	}
 }
@@ -103,11 +108,10 @@ func publishHostPeerUpdate(host *models.Host) error {
 	}
 
 	event := models.Event{
-		ID:    host.ID.String(),
+		ID:    servercfg.GetServer(),
 		Topic: models.EventTopics.SendHostPeerUpdate,
 	}
 	event.Payload.HostPeerUpdate = &peerUpdate
-
 	data, err := json.Marshal(&event)
 	if err != nil {
 		return err