1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
package ports
import (
"log"
"strconv"
v1 "k8s.io/api/core/v1"
)
// ConvertNodePorts converts CSV data to NodePorts map.
func ConvertNodePorts(data [][]string) (map[uint16]string, bool) {
result := make(map[uint16]string)
for _, record := range data {
port, err := strconv.Atoi(record[1])
if err != nil {
log.Printf("Unable to parse port field: %v", err)
continue
}
result[uint16(port)] = record[0]
}
return result, len(result) > 0
}
// FilterXFailNodePorts removes NodePorts expected to fail from map.
func FilterXFailNodePorts(xfails, nodeports map[uint16]string) {
for port, xfailService := range xfails {
service, ok := nodeports[port]
if !ok {
continue
}
if service != xfailService {
continue
}
delete(nodeports, port)
}
}
// FilterNodePorts extracts NodePorts from ServiceList.
func FilterNodePorts(services *v1.ServiceList) (map[uint16]string, bool) {
nodeports := make(map[uint16]string)
for _, service := range services.Items {
for _, port := range service.Spec.Ports {
if port.NodePort != 0 {
nodeports[uint16(port.NodePort)] = service.ObjectMeta.Name
}
}
}
return nodeports, len(nodeports) > 0
}
// FilterIPAddresses extracts IP addresses from NodeList.
// External IP addresses take precedence over internal ones.
func FilterIPAddresses(nodes *v1.NodeList) ([]string, bool) {
addresses := make([]string, 0)
for _, node := range nodes.Items {
for _, address := range node.Status.Addresses {
switch address.Type {
case "InternalIP":
addresses = append(addresses, address.Address)
case "ExternalIP":
addresses = append([]string{address.Address}, addresses...)
}
}
}
return addresses, len(addresses) > 0
}
|
