forked from datahub-project/datahub
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(ui) Update auto-complete functionality and design (datahub-proje…
…ct#7515) Adds in new styles for auto-complete as well as the new functionality around quick filters.
- Loading branch information
1 parent
279f38a
commit ba6426e
Showing
67 changed files
with
1,881 additions
and
261 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
177 changes: 177 additions & 0 deletions
177
.../src/main/java/com/linkedin/datahub/graphql/resolvers/search/GetQuickFiltersResolver.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,177 @@ | ||
| package com.linkedin.datahub.graphql.resolvers.search; | ||
|
|
||
| import com.datahub.authentication.Authentication; | ||
| import com.linkedin.common.urn.UrnUtils; | ||
| import com.linkedin.datahub.graphql.generated.Entity; | ||
| import com.linkedin.datahub.graphql.generated.GetQuickFiltersInput; | ||
| import com.linkedin.datahub.graphql.generated.GetQuickFiltersResult; | ||
| import com.linkedin.datahub.graphql.generated.QuickFilter; | ||
| import com.linkedin.datahub.graphql.resolvers.EntityTypeMapper; | ||
| import com.linkedin.datahub.graphql.resolvers.ResolverUtils; | ||
| import com.linkedin.datahub.graphql.types.common.mappers.UrnToEntityMapper; | ||
| import com.linkedin.entity.client.EntityClient; | ||
| import com.linkedin.metadata.search.AggregationMetadata; | ||
| import com.linkedin.metadata.search.AggregationMetadataArray; | ||
| import com.linkedin.metadata.search.FilterValue; | ||
| import com.linkedin.metadata.search.SearchResult; | ||
| import com.linkedin.metadata.service.ViewService; | ||
| import com.linkedin.view.DataHubViewInfo; | ||
| import graphql.schema.DataFetcher; | ||
| import graphql.schema.DataFetchingEnvironment; | ||
| import lombok.RequiredArgsConstructor; | ||
| import lombok.extern.slf4j.Slf4j; | ||
|
|
||
| import javax.annotation.Nonnull; | ||
| import java.util.ArrayList; | ||
| import java.util.Comparator; | ||
| import java.util.List; | ||
| import java.util.Optional; | ||
| import java.util.concurrent.CompletableFuture; | ||
| import java.util.stream.Collectors; | ||
|
|
||
| import static com.linkedin.datahub.graphql.resolvers.ResolverUtils.bindArgument; | ||
| import static com.linkedin.datahub.graphql.resolvers.search.SearchUtils.SEARCHABLE_ENTITY_TYPES; | ||
| import static com.linkedin.datahub.graphql.resolvers.search.SearchUtils.resolveView; | ||
|
|
||
| @Slf4j | ||
| @RequiredArgsConstructor | ||
| public class GetQuickFiltersResolver implements DataFetcher<CompletableFuture<GetQuickFiltersResult>> { | ||
|
|
||
| private final EntityClient _entityClient; | ||
| private final ViewService _viewService; | ||
|
|
||
| private static final String PLATFORM = "platform"; | ||
| private static final int PLATFORM_COUNT = 5; | ||
| private static final String ENTITY = "entity"; | ||
| private static final int SOURCE_ENTITY_COUNT = 3; | ||
| private static final int DATAHUB_ENTITY_COUNT = 2; | ||
|
|
||
| public CompletableFuture<GetQuickFiltersResult> get(final DataFetchingEnvironment environment) throws Exception { | ||
| final GetQuickFiltersInput input = bindArgument(environment.getArgument("input"), GetQuickFiltersInput.class); | ||
|
|
||
| return CompletableFuture.supplyAsync(() -> { | ||
| final GetQuickFiltersResult result = new GetQuickFiltersResult(); | ||
| final List<QuickFilter> quickFilters = new ArrayList<>(); | ||
|
|
||
| try { | ||
| final SearchResult searchResult = getSearchResults(ResolverUtils.getAuthentication(environment), input); | ||
| final AggregationMetadataArray aggregations = searchResult.getMetadata().getAggregations(); | ||
|
|
||
| quickFilters.addAll(getPlatformQuickFilters(aggregations)); | ||
| quickFilters.addAll(getEntityTypeQuickFilters(aggregations)); | ||
| } catch (Exception e) { | ||
| log.error("Failed getting quick filters", e); | ||
| throw new RuntimeException("Failed to to get quick filters", e); | ||
| } | ||
|
|
||
| result.setQuickFilters(quickFilters); | ||
| return result; | ||
| }); | ||
| } | ||
|
|
||
| /** | ||
| * Do a star search with view filter applied to get info about all data in this instance. | ||
| */ | ||
| private SearchResult getSearchResults(@Nonnull final Authentication authentication, @Nonnull final GetQuickFiltersInput input) throws Exception { | ||
| final DataHubViewInfo maybeResolvedView = (input.getViewUrn() != null) | ||
| ? resolveView(_viewService, UrnUtils.getUrn(input.getViewUrn()), authentication) | ||
| : null; | ||
| final List<String> entityNames = SEARCHABLE_ENTITY_TYPES.stream().map(EntityTypeMapper::getName).collect(Collectors.toList()); | ||
|
|
||
| return _entityClient.searchAcrossEntities( | ||
| maybeResolvedView != null | ||
| ? SearchUtils.intersectEntityTypes(entityNames, maybeResolvedView.getDefinition().getEntityTypes()) | ||
| : entityNames, | ||
| "*", | ||
| maybeResolvedView != null | ||
| ? SearchUtils.combineFilters(null, maybeResolvedView.getDefinition().getFilter()) | ||
| : null, | ||
| 0, | ||
| 0, | ||
| null, | ||
| authentication); | ||
| } | ||
|
|
||
| /** | ||
| * Get platforms and their count from an aggregations array, sorts by entity count, and map the top 5 to quick filters | ||
| */ | ||
| private List<QuickFilter> getPlatformQuickFilters(@Nonnull final AggregationMetadataArray aggregations) { | ||
| final List<QuickFilter> platforms = new ArrayList<>(); | ||
| final Optional<AggregationMetadata> platformAggregations = aggregations.stream().filter(agg -> agg.getName().equals(PLATFORM)).findFirst(); | ||
| if (platformAggregations.isPresent()) { | ||
| final List<FilterValue> sortedPlatforms = | ||
| platformAggregations.get().getFilterValues().stream().sorted(Comparator.comparingLong(val -> -val.getFacetCount())).collect(Collectors.toList()); | ||
| sortedPlatforms.forEach(platformFilter -> { | ||
| if (platforms.size() < PLATFORM_COUNT && platformFilter.getFacetCount() > 0) { | ||
| platforms.add(mapQuickFilter(PLATFORM, platformFilter)); | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| // return platforms sorted alphabetically by their name | ||
| return platforms.stream().sorted(Comparator.comparing(QuickFilter::getValue)).collect(Collectors.toList()); | ||
| } | ||
|
|
||
| /** | ||
| * Gets entity type quick filters from search aggregations. First, get source entity type quick filters | ||
| * from a prioritized list. Do the same for datathub entity types. | ||
| */ | ||
| private List<QuickFilter> getEntityTypeQuickFilters(@Nonnull final AggregationMetadataArray aggregations) { | ||
| final List<QuickFilter> entityTypes = new ArrayList<>(); | ||
| final Optional<AggregationMetadata> entityAggregations = aggregations.stream().filter(agg -> agg.getName().equals(ENTITY)).findFirst(); | ||
|
|
||
| if (entityAggregations.isPresent()) { | ||
| final List<QuickFilter> sourceEntityTypeFilters = | ||
| getQuickFiltersFromList(SearchUtils.PRIORITIZED_SOURCE_ENTITY_TYPES, SOURCE_ENTITY_COUNT, entityAggregations.get()); | ||
| entityTypes.addAll(sourceEntityTypeFilters); | ||
|
|
||
| final List<QuickFilter> dataHubEntityTypeFilters = | ||
| getQuickFiltersFromList(SearchUtils.PRIORITIZED_DATAHUB_ENTITY_TYPES, DATAHUB_ENTITY_COUNT, entityAggregations.get()); | ||
| entityTypes.addAll(dataHubEntityTypeFilters); | ||
| } | ||
| return entityTypes; | ||
| } | ||
|
|
||
| /** | ||
| * Create a quick filters list by looping over prioritized list and adding filters that exist until we reach the maxListSize defined | ||
| */ | ||
| private List<QuickFilter> getQuickFiltersFromList( | ||
| @Nonnull final List<String> prioritizedList, | ||
| final int maxListSize, | ||
| @Nonnull final AggregationMetadata entityAggregations | ||
| ) { | ||
| final List<QuickFilter> entityTypes = new ArrayList<>(); | ||
| prioritizedList.forEach(entityType -> { | ||
| if (entityTypes.size() < maxListSize) { | ||
| final Optional<FilterValue> entityFilter = entityAggregations.getFilterValues().stream().filter(val -> val.getValue().equals(entityType)).findFirst(); | ||
| if (entityFilter.isPresent() && entityFilter.get().getFacetCount() > 0) { | ||
| entityTypes.add(mapQuickFilter(ENTITY, entityFilter.get())); | ||
| } | ||
| } | ||
| }); | ||
|
|
||
| return entityTypes; | ||
| } | ||
|
|
||
| private QuickFilter mapQuickFilter(@Nonnull final String field, @Nonnull final FilterValue filterValue) { | ||
| final boolean isEntityTypeFilter = field.equals(ENTITY); | ||
| final QuickFilter quickFilter = new QuickFilter(); | ||
| quickFilter.setField(field); | ||
| quickFilter.setValue(convertFilterValue(filterValue.getValue(), isEntityTypeFilter)); | ||
| if (filterValue.getEntity() != null) { | ||
| final Entity entity = UrnToEntityMapper.map(filterValue.getEntity()); | ||
| quickFilter.setEntity(entity); | ||
| } | ||
| return quickFilter; | ||
| } | ||
|
|
||
| /** | ||
| * If we're working with an entity type filter, we need to convert the value to an EntityType | ||
| */ | ||
| public static String convertFilterValue(String filterValue, boolean isEntityType) { | ||
| if (isEntityType) { | ||
| return EntityTypeMapper.getType(filterValue).toString(); | ||
| } | ||
| return filterValue; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.