-
Notifications
You must be signed in to change notification settings - Fork 350
/
endpoints.go
111 lines (88 loc) · 2.44 KB
/
endpoints.go
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package kubernetes
import (
"net"
"strconv"
"github.com/zalando/skipper/dataclients/kubernetes/definitions"
)
type endpointID struct {
definitions.ResourceID
TargetPort string
Protocol string
}
type endpoint struct {
Meta *definitions.Metadata `json:"metadata"`
Subsets []*subset `json:"subsets"`
}
type endpointList struct {
Items []*endpoint `json:"items"`
}
func formatEndpointString(ip, protocol string, port int) string {
return protocol + "://" + net.JoinHostPort(ip, strconv.Itoa(port))
}
func formatEndpoint(a *address, p *port, protocol string) string {
return formatEndpointString(a.IP, protocol, p.Port)
}
func formatEndpointsForSubsetAddresses(addresses []*address, port *port, protocol string) []string {
var result []string
for _, address := range addresses {
result = append(result, formatEndpoint(address, port, protocol))
}
return result
}
func (ep *endpoint) targetsByServicePort(protocol string, servicePort *servicePort) []string {
for _, s := range ep.Subsets {
// If only one port exists in the endpoint, use it
if len(s.Ports) == 1 {
return formatEndpointsForSubsetAddresses(s.Addresses, s.Ports[0], protocol)
}
// Otherwise match port by name
for _, p := range s.Ports {
if p.Name != servicePort.Name {
continue
}
return formatEndpointsForSubsetAddresses(s.Addresses, p, protocol)
}
}
return nil
}
func (ep *endpoint) targetsByServiceTarget(protocol string, serviceTarget *definitions.BackendPort) []string {
portName, named := serviceTarget.Value.(string)
portValue, byValue := serviceTarget.Value.(int)
for _, s := range ep.Subsets {
for _, p := range s.Ports {
if named && p.Name != portName || byValue && p.Port != portValue {
continue
}
var result []string
for _, a := range s.Addresses {
result = append(result, formatEndpoint(a, p, protocol))
}
return result
}
}
return nil
}
func (ep *endpoint) targets(protocol string) []string {
result := make([]string, 0)
for _, s := range ep.Subsets {
for _, p := range s.Ports {
for _, a := range s.Addresses {
result = append(result, formatEndpoint(a, p, protocol))
}
}
}
return result
}
type subset struct {
Addresses []*address `json:"addresses"`
Ports []*port `json:"ports"`
}
type address struct {
IP string `json:"ip"`
Node string `json:"nodeName"`
}
type port struct {
Name string `json:"name"`
Port int `json:"port"`
Protocol string `json:"protocol"`
}