Przeglądaj źródła

broadcast update peer

Abhishek Kondur 2 lat temu
rodzic
commit
3ec3d4d02d
3 zmienionych plików z 3 dodań i 2 usunięć
  1. 1 1
      auth/host_session.go
  2. 1 0
      controllers/node.go
  3. 1 1
      mq/publishers.go

+ 1 - 1
auth/host_session.go

@@ -237,7 +237,7 @@ func CheckNetRegAndHostUpdate(networks []string, h *models.Host) {
 				Host:   *h,
 				Node:   *newNode,
 			})
-			mq.BroadCastAddPeer(h, newNode, false)
+			mq.BroadCastAddOrUpdatePeer(h, newNode, false)
 		}
 	}
 	if servercfg.IsMessageQueueBackend() {

+ 1 - 0
controllers/node.go

@@ -699,6 +699,7 @@ func updateNode(w http.ResponseWriter, r *http.Request) {
 				logger.Log(0, "error during node ACL update for node", newNode.ID.String())
 			}
 		}
+		mq.BroadCastAddOrUpdatePeer(host, newNode, true)
 		if err := mq.PublishReplaceDNS(&currentNode, newNode, host); err != nil {
 			logger.Log(1, "failed to publish dns update", err.Error())
 		}

+ 1 - 1
mq/publishers.go

@@ -136,7 +136,7 @@ func BroadCastDelPeer(host *models.Host, network string) error {
 	return nil
 }
 
-func BroadCastAddPeer(host *models.Host, node *models.Node, update bool) error {
+func BroadCastAddOrUpdatePeer(host *models.Host, node *models.Node, update bool) error {
 	nodes, err := logic.GetNetworkNodes(node.Network)
 	if err != nil {
 		return err