func (oi *OsdnRegistryInterface) WatchNodes(receiver chan<- *osdnapi.NodeEvent, ready chan<- bool, start <-chan string, stop <-chan bool) error {
eventQueue, startVersion := oi.createAndRunEventQueue("Node", ready, start)
nodeAddressMap, err := oi.getNodeAddressMap()
if err != nil {
return err
}
checkCondition := true
for {
eventType, obj, err := getEvent(eventQueue, startVersion, &checkCondition)
if err != nil {
return err
}
node := obj.(*kapi.Node)
nodeIP := ""
if len(node.Status.Addresses) > 0 {
nodeIP = node.Status.Addresses[0].Address
} else {
nodeIP, err = osdn.GetNodeIP(node.ObjectMeta.Name)
if err != nil {
return err
}
}
switch eventType {
case watch.Added:
receiver <- &osdnapi.NodeEvent{Type: osdnapi.Added, Node: osdnapi.Node{Name: node.ObjectMeta.Name, IP: nodeIP}}
nodeAddressMap[node.ObjectMeta.UID] = nodeIP
case watch.Modified:
oldNodeIP, ok := nodeAddressMap[node.ObjectMeta.UID]
if ok && oldNodeIP != nodeIP {
// Node Added event will handle update subnet if there is ip mismatch
receiver <- &osdnapi.NodeEvent{Type: osdnapi.Added, Node: osdnapi.Node{Name: node.ObjectMeta.Name, IP: nodeIP}}
nodeAddressMap[node.ObjectMeta.UID] = nodeIP
}
case watch.Deleted:
receiver <- &osdnapi.NodeEvent{Type: osdnapi.Deleted, Node: osdnapi.Node{Name: node.ObjectMeta.Name}}
delete(nodeAddressMap, node.ObjectMeta.UID)
}
}
}
func (oi *OsdnRegistryInterface) GetNodes() ([]osdnapi.Node, string, error) {
knodes, err := oi.kClient.Nodes().List(labels.Everything(), fields.Everything())
if err != nil {
return nil, "", err
}
nodes := make([]osdnapi.Node, 0, len(knodes.Items))
for _, node := range knodes.Items {
var nodeIP string
if len(node.Status.Addresses) > 0 {
nodeIP = node.Status.Addresses[0].Address
} else {
var err error
nodeIP, err = osdn.GetNodeIP(node.ObjectMeta.Name)
if err != nil {
return nil, "", err
}
}
nodes = append(nodes, osdnapi.Node{Name: node.ObjectMeta.Name, IP: nodeIP})
}
return nodes, knodes.ListMeta.ResourceVersion, nil
}