Browse Source

added force

0xdcarns 3 years ago
parent
commit
49e0545274
1 changed files with 11 additions and 7 deletions
  1. 11 7
      controllers/node_grpc.go

+ 11 - 7
controllers/node_grpc.go

@@ -107,12 +107,7 @@ func (s *NodeServiceServer) CreateNode(ctx context.Context, req *nodepb.Object)
 		Type: nodepb.NODE_TYPE,
 	}
 
-	var currentServerNode, getErr = logic.GetNetworkServerLeader(node.Network)
-	if getErr == nil {
-		if err := logic.ServerUpdate(&currentServerNode, false); err != nil {
-			logger.Log(1, "server node:", currentServerNode.ID, "failed update")
-		}
-	}
+	runForceServerUpdate(&node)
 
 	go func(node *models.Node) {
 		if node.UDPHolePunch == "yes" {
@@ -221,7 +216,7 @@ func (s *NodeServiceServer) DeleteNode(ctx context.Context, req *nodepb.Object)
 		return nil, err
 	}
 
-	runUpdates(&node, false)
+	runForceServerUpdate(&node)
 
 	return &nodepb.Object{
 		Data: "success",
@@ -313,3 +308,12 @@ func getNodeFromRequestData(data string) (models.Node, error) {
 func isServer(node *models.Node) bool {
 	return node.IsServer == "yes"
 }
+
+func runForceServerUpdate(node *models.Node) {
+	var currentServerNode, getErr = logic.GetNetworkServerLeader(node.Network)
+	if getErr == nil {
+		if err := logic.ServerUpdate(&currentServerNode, false); err != nil {
+			logger.Log(1, "server node:", currentServerNode.ID, "failed update")
+		}
+	}
+}