From ffea85d3978600a86815273d37f8bfb51f57a429 Mon Sep 17 00:00:00 2001 From: Manuel de Brito Fontes Date: Thu, 18 Jan 2018 17:35:00 -0300 Subject: [PATCH] Cleanup --- internal/ingress/controller/store/endpoint.go | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/internal/ingress/controller/store/endpoint.go b/internal/ingress/controller/store/endpoint.go index c464e98b5..126b616b6 100644 --- a/internal/ingress/controller/store/endpoint.go +++ b/internal/ingress/controller/store/endpoint.go @@ -30,11 +30,13 @@ type EndpointLister struct { // GetServiceEndpoints returns the endpoints of a service, matched on service name. func (s *EndpointLister) GetServiceEndpoints(svc *apiv1.Service) (*apiv1.Endpoints, error) { - for _, m := range s.Store.List() { - ep := m.(*apiv1.Endpoints) - if svc.Name == ep.Name && svc.Namespace == ep.Namespace { - return ep, nil - } + key := fmt.Sprintf("%v/%v", svc.Namespace, svc.Name) + eps, exists, err := s.GetByKey(key) + if err != nil { + return nil, err } - return nil, fmt.Errorf("could not find endpoints for service: %v", svc.Name) + if !exists { + return nil, fmt.Errorf("could not find endpoints for service %v", key) + } + return eps.(*apiv1.Endpoints), nil }