diff --git a/internal/resource/mappers/bimapper/bimapper.go b/internal/resource/mappers/bimapper/bimapper.go index 62a6b4dc7ffe..b17381306737 100644 --- a/internal/resource/mappers/bimapper/bimapper.go +++ b/internal/resource/mappers/bimapper/bimapper.go @@ -272,7 +272,10 @@ func (m *Mapper) MapLink(_ context.Context, _ controller.Runtime, res *pbresourc } func (m *Mapper) itemIDsByLink(link resource.ReferenceKey) []*pbresource.ID { - items, ok := m.getItemsByLink(link) + m.lock.Lock() + defer m.lock.Unlock() + + items, ok := m.linkToItem[link] if !ok { return nil } @@ -285,7 +288,10 @@ func (m *Mapper) itemIDsByLink(link resource.ReferenceKey) []*pbresource.ID { } func (m *Mapper) itemRefsByLink(link resource.ReferenceKey) []*pbresource.Reference { - items, ok := m.getItemsByLink(link) + m.lock.Lock() + defer m.lock.Unlock() + + items, ok := m.linkToItem[link] if !ok { return nil } @@ -296,11 +302,3 @@ func (m *Mapper) itemRefsByLink(link resource.ReferenceKey) []*pbresource.Refere } return out } - -func (m *Mapper) getItemsByLink(link resource.ReferenceKey) (map[resource.ReferenceKey]struct{}, bool) { - m.lock.Lock() - defer m.lock.Unlock() - - items, ok := m.linkToItem[link] - return items, ok -}