-
Notifications
You must be signed in to change notification settings - Fork 3k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(recommendations): Recommendations infra P1 (#3455)
- Loading branch information
1 parent
87e93ab
commit 2f03ad8
Showing
107 changed files
with
4,334 additions
and
305 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
165 changes: 165 additions & 0 deletions
165
...va/com/linkedin/datahub/graphql/resolvers/recommendation/ListRecommendationsResolver.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,165 @@ | ||
| package com.linkedin.datahub.graphql.resolvers.recommendation; | ||
|
|
||
| import com.linkedin.common.urn.Urn; | ||
| import com.linkedin.datahub.graphql.generated.ContentParams; | ||
| import com.linkedin.datahub.graphql.generated.EntityProfileParams; | ||
| import com.linkedin.datahub.graphql.generated.Filter; | ||
| import com.linkedin.datahub.graphql.generated.ListRecommendationsInput; | ||
| import com.linkedin.datahub.graphql.generated.ListRecommendationsResult; | ||
| import com.linkedin.datahub.graphql.generated.RecommendationContent; | ||
| import com.linkedin.datahub.graphql.generated.RecommendationModule; | ||
| import com.linkedin.datahub.graphql.generated.RecommendationParams; | ||
| import com.linkedin.datahub.graphql.generated.RecommendationRenderType; | ||
| import com.linkedin.datahub.graphql.generated.RecommendationRequestContext; | ||
| import com.linkedin.datahub.graphql.generated.SearchParams; | ||
| import com.linkedin.datahub.graphql.resolvers.EntityTypeMapper; | ||
| import com.linkedin.datahub.graphql.types.common.mappers.UrnToEntityMapper; | ||
| import com.linkedin.metadata.query.filter.Criterion; | ||
| import com.linkedin.metadata.query.filter.CriterionArray; | ||
| import com.linkedin.metadata.recommendation.EntityRequestContext; | ||
| import com.linkedin.metadata.recommendation.RecommendationsService; | ||
| import com.linkedin.metadata.recommendation.SearchRequestContext; | ||
| import graphql.schema.DataFetcher; | ||
| import graphql.schema.DataFetchingEnvironment; | ||
| import java.net.URISyntaxException; | ||
| import java.util.Collections; | ||
| import java.util.List; | ||
| import java.util.Optional; | ||
| import java.util.concurrent.CompletableFuture; | ||
| import java.util.stream.Collectors; | ||
| import lombok.RequiredArgsConstructor; | ||
| import lombok.extern.slf4j.Slf4j; | ||
|
|
||
| import static com.linkedin.datahub.graphql.resolvers.ResolverUtils.bindArgument; | ||
|
|
||
|
|
||
| @Slf4j | ||
| @RequiredArgsConstructor | ||
| public class ListRecommendationsResolver implements DataFetcher<CompletableFuture<ListRecommendationsResult>> { | ||
|
|
||
| private static final ListRecommendationsResult EMPTY_RECOMMENDATIONS = new ListRecommendationsResult(Collections.emptyList()); | ||
|
|
||
| private final RecommendationsService _recommendationsService; | ||
|
|
||
| @Override | ||
| public CompletableFuture<ListRecommendationsResult> get(DataFetchingEnvironment environment) { | ||
| final ListRecommendationsInput input = | ||
| bindArgument(environment.getArgument("input"), ListRecommendationsInput.class); | ||
|
|
||
| return CompletableFuture.supplyAsync(() -> { | ||
| try { | ||
| log.debug("Listing recommendations for input {}", input); | ||
| List<com.linkedin.metadata.recommendation.RecommendationModule> modules = | ||
| _recommendationsService.listRecommendations(Urn.createFromString(input.getUserUrn()), | ||
| mapRequestContext(input.getRequestContext()), input.getLimit()); | ||
| return ListRecommendationsResult.builder() | ||
| .setModules(modules.stream() | ||
| .map(this::mapRecommendationModule) | ||
| .filter(Optional::isPresent) | ||
| .map(Optional::get) | ||
| .collect(Collectors.toList())) | ||
| .build(); | ||
| } catch (Exception e) { | ||
| log.error("Failed to get recommendations for input {}", input, e); | ||
| return EMPTY_RECOMMENDATIONS; | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| private com.linkedin.metadata.recommendation.RecommendationRequestContext mapRequestContext( | ||
| RecommendationRequestContext requestContext) { | ||
| com.linkedin.metadata.recommendation.ScenarioType mappedScenarioType; | ||
| try { | ||
| mappedScenarioType = | ||
| com.linkedin.metadata.recommendation.ScenarioType.valueOf(requestContext.getScenario().toString()); | ||
| } catch (IllegalArgumentException e) { | ||
| log.error("Failed to map scenario type: {}", requestContext.getScenario(), e); | ||
| throw e; | ||
| } | ||
| com.linkedin.metadata.recommendation.RecommendationRequestContext mappedRequestContext = | ||
| new com.linkedin.metadata.recommendation.RecommendationRequestContext().setScenario(mappedScenarioType); | ||
| if (requestContext.getSearchRequestContext() != null) { | ||
| SearchRequestContext searchRequestContext = | ||
| new SearchRequestContext().setQuery(requestContext.getSearchRequestContext().getQuery()); | ||
| if (requestContext.getSearchRequestContext().getFilters() != null) { | ||
| searchRequestContext.setFilters(new CriterionArray(requestContext.getSearchRequestContext() | ||
| .getFilters() | ||
| .stream() | ||
| .map(facetField -> new Criterion().setField(facetField.getField()).setValue(facetField.getValue())) | ||
| .collect(Collectors.toList()))); | ||
| } | ||
| mappedRequestContext.setSearchRequestContext(searchRequestContext); | ||
| } | ||
| if (requestContext.getEntityRequestContext() != null) { | ||
| Urn entityUrn; | ||
| try { | ||
| entityUrn = Urn.createFromString(requestContext.getEntityRequestContext().getUrn()); | ||
| } catch (URISyntaxException e) { | ||
| log.error("Malformed URN while mapping recommendations request: {}", | ||
| requestContext.getEntityRequestContext().getUrn(), e); | ||
| throw new IllegalArgumentException(e); | ||
| } | ||
| EntityRequestContext entityRequestContext = new EntityRequestContext().setUrn(entityUrn) | ||
| .setType(EntityTypeMapper.getName(requestContext.getEntityRequestContext().getType())); | ||
| mappedRequestContext.setEntityRequestContext(entityRequestContext); | ||
| } | ||
| return mappedRequestContext; | ||
| } | ||
|
|
||
| private Optional<RecommendationModule> mapRecommendationModule( | ||
| com.linkedin.metadata.recommendation.RecommendationModule module) { | ||
| RecommendationModule mappedModule = new RecommendationModule(); | ||
| mappedModule.setTitle(module.getTitle()); | ||
| mappedModule.setModuleId(module.getModuleId()); | ||
| try { | ||
| mappedModule.setRenderType(RecommendationRenderType.valueOf(module.getRenderType().toString())); | ||
| } catch (IllegalArgumentException e) { | ||
| log.error("Failed to map render type: {}", module.getRenderType(), e); | ||
| throw e; | ||
| } | ||
| mappedModule.setContent( | ||
| module.getContent().stream().map(this::mapRecommendationContent).collect(Collectors.toList())); | ||
| return Optional.of(mappedModule); | ||
| } | ||
|
|
||
| private RecommendationContent mapRecommendationContent( | ||
| com.linkedin.metadata.recommendation.RecommendationContent content) { | ||
| RecommendationContent mappedContent = new RecommendationContent(); | ||
| mappedContent.setValue(content.getValue()); | ||
| if (content.hasEntity()) { | ||
| mappedContent.setEntity(UrnToEntityMapper.map(content.getEntity())); | ||
| } | ||
| if (content.hasParams()) { | ||
| mappedContent.setParams(mapRecommendationParams(content.getParams())); | ||
| } | ||
| return mappedContent; | ||
| } | ||
|
|
||
| private RecommendationParams mapRecommendationParams( | ||
| com.linkedin.metadata.recommendation.RecommendationParams params) { | ||
| RecommendationParams mappedParams = new RecommendationParams(); | ||
| if (params.hasSearchParams()) { | ||
| SearchParams searchParams = new SearchParams(); | ||
| searchParams.setQuery(params.getSearchParams().getQuery()); | ||
| if (!params.getSearchParams().getFilters().isEmpty()) { | ||
| searchParams.setFilters(params.getSearchParams() | ||
| .getFilters() | ||
| .stream() | ||
| .map(criterion -> Filter.builder().setField(criterion.getField()).setValue(criterion.getValue()).build()) | ||
| .collect(Collectors.toList())); | ||
| } | ||
| mappedParams.setSearchParams(searchParams); | ||
| } | ||
|
|
||
| if (params.hasEntityProfileParams()) { | ||
| mappedParams.setEntityProfileParams( | ||
| EntityProfileParams.builder().setUrn(params.getEntityProfileParams().getUrn().toString()).build()); | ||
| } | ||
|
|
||
| if (params.hasContentParams()) { | ||
| mappedParams.setContentParams(ContentParams.builder().setCount(params.getContentParams().getCount()).build()); | ||
| } | ||
|
|
||
| return mappedParams; | ||
| } | ||
| } |
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.