fix ipset on pod creation/deletion

kubeovn · Aug 25, 2021 · c57c6db · c57c6db
1 parent ef9dbc5
commit c57c6db
Showing 1 changed file with 24 additions and 14 deletions.
diff --git a/pkg/daemon/gateway.go b/pkg/daemon/gateway.go
@@ -182,7 +182,8 @@ func (c *Controller) addEgressConfig(subnet, ip string) error {
 		return err
 	}
 
-	if podSubnet.Spec.GatewayType != kubeovnv1.GWDistributedType ||
+	if podSubnet.Spec.Vlan != "" ||
+		podSubnet.Spec.GatewayType != kubeovnv1.GWDistributedType ||
 		podSubnet.Spec.Vpc != util.DefaultVpc {
 		return nil
 	}
@@ -211,7 +212,8 @@ func (c *Controller) removeEgressConfig(subnet, ip string) error {
 		return err
 	}
 
-	if podSubnet.Spec.GatewayType != kubeovnv1.GWDistributedType ||
+	if podSubnet.Spec.Vlan != "" ||
+		podSubnet.Spec.GatewayType != kubeovnv1.GWDistributedType ||
 		podSubnet.Spec.Vpc != util.DefaultVpc {
 		return nil
 	}
@@ -231,24 +233,32 @@ func (c *Controller) removeEgressConfig(subnet, ip string) error {
 
 func (c *Controller) addIPSetMembers(setID, protocol string, ips []string) {
 	if protocol == kubeovnv1.ProtocolDual {
-		c.ipset[kubeovnv1.ProtocolIPv4].AddMembers(setID, []string{ips[0]})
-		c.ipset[kubeovnv1.ProtocolIPv6].AddMembers(setID, []string{ips[1]})
-		c.ipset[kubeovnv1.ProtocolIPv4].ApplyUpdates()
-		c.ipset[kubeovnv1.ProtocolIPv6].ApplyUpdates()
-	} else {
-		c.ipset[protocol].AddMembers(setID, []string{ips[0]})
+		if c.ipset[kubeovnv1.ProtocolIPv4] != nil {
+			c.ipset[kubeovnv1.ProtocolIPv4].AddMembers(setID, ips[:1])
+			c.ipset[kubeovnv1.ProtocolIPv4].ApplyUpdates()
+		}
+		if c.ipset[kubeovnv1.ProtocolIPv6] != nil {
+			c.ipset[kubeovnv1.ProtocolIPv6].AddMembers(setID, ips[1:])
+			c.ipset[kubeovnv1.ProtocolIPv6].ApplyUpdates()
+		}
+	} else if c.ipset[protocol] != nil {
+		c.ipset[protocol].AddMembers(setID, ips[:1])
 		c.ipset[protocol].ApplyUpdates()
 	}
 }
 
 func (c *Controller) removeIPSetMembers(setID, protocol string, ips []string) {
 	if protocol == kubeovnv1.ProtocolDual {
-		c.ipset[kubeovnv1.ProtocolIPv4].RemoveMembers(setID, []string{ips[0]})
-		c.ipset[kubeovnv1.ProtocolIPv6].RemoveMembers(setID, []string{ips[1]})
-		c.ipset[kubeovnv1.ProtocolIPv4].ApplyUpdates()
-		c.ipset[kubeovnv1.ProtocolIPv6].ApplyUpdates()
-	} else {
-		c.ipset[protocol].RemoveMembers(setID, []string{ips[0]})
+		if c.ipset[kubeovnv1.ProtocolIPv4] != nil {
+			c.ipset[kubeovnv1.ProtocolIPv4].RemoveMembers(setID, ips[:1])
+			c.ipset[kubeovnv1.ProtocolIPv4].ApplyUpdates()
+		}
+		if c.ipset[kubeovnv1.ProtocolIPv6] != nil {
+			c.ipset[kubeovnv1.ProtocolIPv6].RemoveMembers(setID, ips[1:])
+			c.ipset[kubeovnv1.ProtocolIPv6].ApplyUpdates()
+		}
+	} else if c.ipset[protocol] != nil {
+		c.ipset[protocol].RemoveMembers(setID, ips[:1])
 		c.ipset[protocol].ApplyUpdates()
 	}
 }