697ea476e2
If different pods with same address are exposed by the same service if some of the endpointslices endpoints are overwriten. This change add the pod name to the hash function to ensure that all the endpoints are in place. Signed-off-by: Enrique Llorente <ellorent@redhat.com>
103 lines
2.8 KiB
Go
103 lines
2.8 KiB
Go
/*
|
|
Copyright 2019 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package endpointslice
|
|
|
|
import (
|
|
"sort"
|
|
|
|
discovery "k8s.io/api/discovery/v1"
|
|
endpointutil "k8s.io/kubernetes/pkg/controller/util/endpoint"
|
|
)
|
|
|
|
// endpointHash is used to uniquely identify endpoints. Only including addresses
|
|
// and hostnames as unique identifiers allows us to do more in place updates
|
|
// should attributes such as topology or conditions change.
|
|
type endpointHash string
|
|
type endpointHashObj struct {
|
|
Addresses []string
|
|
Hostname string
|
|
Namespace string
|
|
Name string
|
|
}
|
|
|
|
func hashEndpoint(endpoint *discovery.Endpoint) endpointHash {
|
|
sort.Strings(endpoint.Addresses)
|
|
hashObj := endpointHashObj{Addresses: endpoint.Addresses}
|
|
if endpoint.Hostname != nil {
|
|
hashObj.Hostname = *endpoint.Hostname
|
|
}
|
|
if endpoint.TargetRef != nil {
|
|
hashObj.Namespace = endpoint.TargetRef.Namespace
|
|
hashObj.Name = endpoint.TargetRef.Name
|
|
}
|
|
|
|
return endpointHash(endpointutil.DeepHashObjectToString(hashObj))
|
|
}
|
|
|
|
// EndpointSet provides simple methods for comparing sets of Endpoints.
|
|
type EndpointSet map[endpointHash]*discovery.Endpoint
|
|
|
|
// Insert adds items to the set.
|
|
func (s EndpointSet) Insert(items ...*discovery.Endpoint) EndpointSet {
|
|
for _, item := range items {
|
|
s[hashEndpoint(item)] = item
|
|
}
|
|
return s
|
|
}
|
|
|
|
// Delete removes all items from the set.
|
|
func (s EndpointSet) Delete(items ...*discovery.Endpoint) EndpointSet {
|
|
for _, item := range items {
|
|
delete(s, hashEndpoint(item))
|
|
}
|
|
return s
|
|
}
|
|
|
|
// Has returns true if and only if item is contained in the set.
|
|
func (s EndpointSet) Has(item *discovery.Endpoint) bool {
|
|
_, contained := s[hashEndpoint(item)]
|
|
return contained
|
|
}
|
|
|
|
// Returns an endpoint matching the hash if contained in the set.
|
|
func (s EndpointSet) Get(item *discovery.Endpoint) *discovery.Endpoint {
|
|
return s[hashEndpoint(item)]
|
|
}
|
|
|
|
// UnsortedList returns the slice with contents in random order.
|
|
func (s EndpointSet) UnsortedList() []*discovery.Endpoint {
|
|
endpoints := make([]*discovery.Endpoint, 0, len(s))
|
|
for _, endpoint := range s {
|
|
endpoints = append(endpoints, endpoint)
|
|
}
|
|
return endpoints
|
|
}
|
|
|
|
// Returns a single element from the set.
|
|
func (s EndpointSet) PopAny() (*discovery.Endpoint, bool) {
|
|
for _, endpoint := range s {
|
|
s.Delete(endpoint)
|
|
return endpoint, true
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
// Len returns the size of the set.
|
|
func (s EndpointSet) Len() int {
|
|
return len(s)
|
|
}
|