mirror of
https://github.com/cloudnativelabs/kube-router.git
synced 2025-10-08 00:11:07 +02:00
342ea5ac9f
* Move getNodeIP logic to utils package Remove redundant ipset lookups utils.NewIPSet() does this for us. * Don't masquerade pod -> nodeAddrsIPSet traffic Previously with Pod egress enabled, this would get masqueraded. This change also adds cleanup for said ipset. * Enhanced cleanup of Pod egress, overlay networking - Delete old/bad pod egress iptables rule(s) from old versions - When pod egress or overlay are disabled, cleanup as needed * Update IPSet.Sets to map type * ipset enhancements - Avoid providing method that would delete all ipset sets on a system - New method DestroyAllWithin() destroys sets tracked by an IPSet - Create() now handles cases where Sets/System state are not in sync - Refresh() now handles leftover -temp set gracefully - Swap() now uses ipset swap - Delete() improved sync of Sets and system state - Get() now validates if map element exists before trying - etc * Update routes controller to reflect ipset changes
62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
package utils
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes"
|
|
apiv1 "k8s.io/client-go/pkg/api/v1"
|
|
)
|
|
|
|
// GetNodeObject returns the node API object for the node
|
|
func GetNodeObject(clientset *kubernetes.Clientset, hostnameOverride string) (*apiv1.Node, error) {
|
|
|
|
// assuming kube-router is running as pod, first check env NODE_NAME
|
|
nodeName := os.Getenv("NODE_NAME")
|
|
if nodeName != "" {
|
|
node, err := clientset.Core().Nodes().Get(nodeName, metav1.GetOptions{})
|
|
if err == nil {
|
|
return node, nil
|
|
}
|
|
}
|
|
|
|
// if env NODE_NAME is not set then check if node is register with hostname
|
|
hostName, _ := os.Hostname()
|
|
node, err := clientset.Core().Nodes().Get(hostName, metav1.GetOptions{})
|
|
if err == nil {
|
|
return node, nil
|
|
}
|
|
|
|
// if env NODE_NAME is not set and node is not registered with hostname, then use host name override
|
|
if hostnameOverride != "" {
|
|
node, err = clientset.Core().Nodes().Get(hostnameOverride, metav1.GetOptions{})
|
|
if err == nil {
|
|
return node, nil
|
|
}
|
|
}
|
|
|
|
return nil, fmt.Errorf("Failed to identify the node by NODE_NAME, hostname or --hostname-override")
|
|
}
|
|
|
|
// GetNodeIP returns the most valid external facing IP address for a node.
|
|
// Order of preference:
|
|
// 1. NodeInternalIP
|
|
// 2. NodeExternalIP (Only set on cloud providers usually)
|
|
func GetNodeIP(node *apiv1.Node) (net.IP, error) {
|
|
addresses := node.Status.Addresses
|
|
addressMap := make(map[apiv1.NodeAddressType][]apiv1.NodeAddress)
|
|
for i := range addresses {
|
|
addressMap[addresses[i].Type] = append(addressMap[addresses[i].Type], addresses[i])
|
|
}
|
|
if addresses, ok := addressMap[apiv1.NodeInternalIP]; ok {
|
|
return net.ParseIP(addresses[0].Address), nil
|
|
}
|
|
if addresses, ok := addressMap[apiv1.NodeExternalIP]; ok {
|
|
return net.ParseIP(addresses[0].Address), nil
|
|
}
|
|
return nil, errors.New("host IP unknown")
|
|
}
|