Browse Source

update network model

Abhishek Kondur 2 years ago
parent
commit
92922f17bd
3 changed files with 9 additions and 2 deletions
  1. 1 1
      logic/peers.go
  2. 1 1
      models/mqtt.go
  3. 7 0
      models/node.go

+ 1 - 1
logic/peers.go

@@ -478,7 +478,7 @@ func GetPeerUpdateForHost(host *models.Host) (models.HostPeerUpdate, error) {
 			}
 			}
 			hostPeerUpdate.EgressInfo[node.ID.String()] = models.EgressInfo{
 			hostPeerUpdate.EgressInfo[node.ID.String()] = models.EgressInfo{
 				EgressID: node.ID.String(),
 				EgressID: node.ID.String(),
-				Network:  node.Network,
+				Network:  node.PrimaryNetworkRange(),
 				EgressGwAddr: net.IPNet{
 				EgressGwAddr: net.IPNet{
 					IP:   net.ParseIP(node.PrimaryAddress()),
 					IP:   net.ParseIP(node.PrimaryAddress()),
 					Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),
 					Mask: getCIDRMaskFromAddr(node.PrimaryAddress()),

+ 1 - 1
models/mqtt.go

@@ -38,7 +38,7 @@ type IngressInfo struct {
 
 
 type EgressInfo struct {
 type EgressInfo struct {
 	EgressID     string                   `json:"egress_id" yaml:"egress_id"`
 	EgressID     string                   `json:"egress_id" yaml:"egress_id"`
-	Network      string                   `json:"network" yaml:"network"`
+	Network      net.IPNet                `json:"network" yaml:"network"`
 	EgressGwAddr net.IPNet                `json:"egress_gw_addr" yaml:"egress_gw_addr"`
 	EgressGwAddr net.IPNet                `json:"egress_gw_addr" yaml:"egress_gw_addr"`
 	GwPeers      map[string]PeerRouteInfo `json:"gateway_peers" yaml:"gateway_peers"`
 	GwPeers      map[string]PeerRouteInfo `json:"gateway_peers" yaml:"gateway_peers"`
 	EgressGWCfg  EgressGatewayRequest     `json:"egress_gateway_cfg" yaml:"egress_gateway_cfg"`
 	EgressGWCfg  EgressGatewayRequest     `json:"egress_gateway_cfg" yaml:"egress_gateway_cfg"`

+ 7 - 0
models/node.go

@@ -195,6 +195,13 @@ func (node *Node) PrimaryAddress() string {
 	return node.Address6.IP.String()
 	return node.Address6.IP.String()
 }
 }
 
 
+func (node *Node) PrimaryNetworkRange() net.IPNet {
+	if node.NetworkRange.IP != nil {
+		return node.NetworkRange
+	}
+	return node.NetworkRange6
+}
+
 // Node.SetDefaultConnected
 // Node.SetDefaultConnected
 func (node *Node) SetDefaultConnected() {
 func (node *Node) SetDefaultConnected() {
 	node.Connected = true
 	node.Connected = true