Browse Source

add inet info to peer update

abhishek9686 1 year ago
parent
commit
ce19d3e677
2 changed files with 4 additions and 0 deletions
  1. 3 0
      logic/peers.go
  2. 1 0
      models/mqtt.go

+ 3 - 0
logic/peers.go

@@ -143,6 +143,9 @@ func GetPeerUpdateForHost(network string, host *models.Host, allNodes []models.N
 			continue
 			continue
 		}
 		}
 		hostPeerUpdate = SetDefaultGw(node, hostPeerUpdate)
 		hostPeerUpdate = SetDefaultGw(node, hostPeerUpdate)
+		if !hostPeerUpdate.IsInternetGw {
+			hostPeerUpdate.IsInternetGw = IsInternetGw(node)
+		}
 		currentPeers := GetNetworkNodesMemory(allNodes, node.Network)
 		currentPeers := GetNetworkNodesMemory(allNodes, node.Network)
 		for _, peer := range currentPeers {
 		for _, peer := range currentPeers {
 			peer := peer
 			peer := peer

+ 1 - 0
models/mqtt.go

@@ -11,6 +11,7 @@ type HostPeerUpdate struct {
 	Host            Host                 `json:"host" bson:"host" yaml:"host"`
 	Host            Host                 `json:"host" bson:"host" yaml:"host"`
 	ChangeDefaultGw bool                 `json:"change_default_gw"`
 	ChangeDefaultGw bool                 `json:"change_default_gw"`
 	DefaultGwIp     net.IP               `json:"default_gw_ip"`
 	DefaultGwIp     net.IP               `json:"default_gw_ip"`
+	IsInternetGw    bool                 `json:"is_inet_gw"`
 	NodeAddrs       []net.IPNet          `json:"nodes_addrs" yaml:"nodes_addrs"`
 	NodeAddrs       []net.IPNet          `json:"nodes_addrs" yaml:"nodes_addrs"`
 	Server          string               `json:"server" bson:"server" yaml:"server"`
 	Server          string               `json:"server" bson:"server" yaml:"server"`
 	ServerVersion   string               `json:"serverversion" bson:"serverversion" yaml:"serverversion"`
 	ServerVersion   string               `json:"serverversion" bson:"serverversion" yaml:"serverversion"`