yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-12-12 02:25:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add recent searches filtering (#4476)

Browse Source
This commit is contained in:
Dexter Lee 2022-03-23 16:24:19 -07:00 committed by GitHub
parent d04092e634
commit a1ebc6be86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
metadata-io/src/main/java/com/linkedin/metadata/recommendation/candidatesource/RecentlySearchedSource.java
View File

@ -84,6 +84,7 @@ public class RecentlySearchedSource implements RecommendationSource {
.map(bucket -> buildContent(bucket.getKeyAsString()))
.filter(Optional::isPresent)
.map(Optional::get)
.limit(MAX_CONTENT)
.collect(Collectors.toList());
} catch (Exception e) {
log.error("Search query to get most recently viewed entities failed", e);
@ -107,7 +108,7 @@ public class RecentlySearchedSource implements RecommendationSource {
String lastSearched = "last_searched";
AggregationBuilder aggregation = AggregationBuilders.terms(ENTITY_AGG_NAME)
.field(DataHubUsageEventConstants.QUERY + ".keyword")
.size(MAX_CONTENT)
.size(MAX_CONTENT * 2) // Fetch more than max to account for post-filtering
.order(BucketOrder.aggregation(lastSearched, false))
.subAggregation(AggregationBuilders.max(lastSearched).field(DataHubUsageEventConstants.TIMESTAMP));
source.aggregation(aggregation);
@ -118,7 +119,14 @@ public class RecentlySearchedSource implements RecommendationSource {
return request;
}
private boolean isQueryInvalid(@Nonnull String query) {
return query.trim().isEmpty() || query.equals("*");
}
private Optional<RecommendationContent> buildContent(@Nonnull String query) {
if (isQueryInvalid(query)) {
return Optional.empty();
}
return Optional.of(new RecommendationContent().setValue(query)
.setParams(new RecommendationParams().setSearchParams(new SearchParams().setQuery(query))));
}