yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-11-11 08:52:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(lineage): Apply search flags to scroll query in LineageSearchService (#8518)

Browse Source 
Co-authored-by: Indy Prentice <indy@Indys-MacBook-Pro.local>
This commit is contained in:
Indy Prentice 2023-07-29 04:56:49 -03:00 committed by GitHub
parent 4b32e9a9ab
commit e09a2e6c85
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
metadata-io/src/main/java/com/linkedin/metadata/search/LineageSearchService.java
View File

@ -606,6 +606,7 @@ public class LineageSearchService {
private LineageScrollResult getScrollResultInBatches(List<LineageRelationship> lineageRelationships, private LineageScrollResult getScrollResultInBatches(List<LineageRelationship> lineageRelationships,
@Nonnull String input, @Nullable Filter inputFilters, @Nullable SortCriterion sortCriterion, @Nullable String scrollId, @Nonnull String input, @Nullable Filter inputFilters, @Nullable SortCriterion sortCriterion, @Nullable String scrollId,
@Nonnull String keepAlive, int size, @Nonnull SearchFlags searchFlags) { @Nonnull String keepAlive, int size, @Nonnull SearchFlags searchFlags) {
final SearchFlags finalFlags = applyDefaultSearchFlags(searchFlags, input, DEFAULT_SERVICE_SEARCH_FLAGS);
LineageScrollResult finalResult = LineageScrollResult finalResult =
new LineageScrollResult().setEntities(new LineageSearchEntityArray(Collections.emptyList())) new LineageScrollResult().setEntities(new LineageSearchEntityArray(Collections.emptyList()))
.setMetadata(new SearchResultMetadata().setAggregations(new AggregationMetadataArray())) .setMetadata(new SearchResultMetadata().setAggregations(new AggregationMetadataArray()))
@ -623,7 +624,7 @@ public class LineageSearchService {
LineageScrollResult resultForBatch = buildLineageScrollResult( LineageScrollResult resultForBatch = buildLineageScrollResult(
_searchService.scrollAcrossEntities(entitiesToQuery, input, finalFilter, sortCriterion, scrollId, keepAlive, querySize, _searchService.scrollAcrossEntities(entitiesToQuery, input, finalFilter, sortCriterion, scrollId, keepAlive, querySize,
searchFlags), urnToRelationship); finalFlags), urnToRelationship);
querySize = Math.max(0, size - resultForBatch.getEntities().size()); querySize = Math.max(0, size - resultForBatch.getEntities().size());
finalResult = mergeScrollResult(finalResult, resultForBatch); finalResult = mergeScrollResult(finalResult, resultForBatch);
} }