diff --git a/restli-resources/src/main/java/com/linkedin/metadata/restli/BaseSearchableEntityResource.java b/restli-resources/src/main/java/com/linkedin/metadata/restli/BaseSearchableEntityResource.java index 6a264c669..5bbcd16a3 100644 --- a/restli-resources/src/main/java/com/linkedin/metadata/restli/BaseSearchableEntityResource.java +++ b/restli-resources/src/main/java/com/linkedin/metadata/restli/BaseSearchableEntityResource.java @@ -217,9 +217,9 @@ public CollectionResult getSearchQueryCollectionRes final Map urnValueMap = getInternalNonEmpty(matchedUrns, parseAspectsParam(aspectNames, isInternalModelsEnabled), isInternalModelsEnabled); - final List existingUrns = matchedUrns.stream().filter(urn -> urnValueMap.containsKey(urn)).collect(Collectors.toList()); + final List existingUrns = matchedUrns.stream().filter(urnValueMap::containsKey).collect(Collectors.toList()); return new CollectionResult<>( - existingUrns.stream().map(urn -> urnValueMap.get(urn)).collect(Collectors.toList()), + existingUrns.stream().map(urnValueMap::get).collect(Collectors.toList()), searchResult.getTotalCount(), searchResult.getSearchResultMetadata().setUrns(new UrnArray(existingUrns.stream().map(urn -> (Urn) urn).collect(Collectors.toList()))) );