Browse Source

unset inet gw fix

abhishek9686 1 năm trước cách đây
mục cha
commit
8633821705
2 tập tin đã thay đổi với 3 bổ sung8 xóa
  1. 2 7
      logic/gateway.go
  2. 1 1
      pro/logic/nodes.go

+ 2 - 7
logic/gateway.go

@@ -22,13 +22,8 @@ func GetInternetGateways() ([]models.Node, error) {
 	}
 	igs := make([]models.Node, 0)
 	for _, node := range nodes {
-		if !node.IsEgressGateway {
-			continue
-		}
-		for _, ran := range node.EgressGatewayRanges {
-			if ran == "0.0.0.0/0" {
-				igs = append(igs, node)
-			}
+		if node.IsInternetGateway {
+			igs = append(igs, node)
 		}
 	}
 	return igs, nil

+ 1 - 1
pro/logic/nodes.go

@@ -68,7 +68,7 @@ func UnsetInternetGw(node *models.Node) {
 	for _, clientNode := range nodes {
 		if node.ID.String() == clientNode.InternetGwID {
 			node.InternetGwID = ""
-			logic.UpsertNode(node)
+			logic.UpsertNode(&clientNode)
 		}
 
 	}