use of org.finra.herd.model.dto.BusinessObjectDefinitionIndexSearchResponseDto in project herd by FINRAOS.
the class BusinessObjectDefinitionServiceImpl method indexSearchBusinessObjectDefinitions.
@Override
public BusinessObjectDefinitionIndexSearchResponse indexSearchBusinessObjectDefinitions(BusinessObjectDefinitionIndexSearchRequest searchRequest, Set<String> fieldsRequested) {
// Get the configured values for index name and document type
final String indexName = searchIndexDaoHelper.getActiveSearchIndex(SearchIndexTypeEntity.SearchIndexTypes.BUS_OBJCT_DFNTN.name());
final String documentType = configurationHelper.getProperty(ConfigurationValue.ELASTICSEARCH_BDEF_DOCUMENT_TYPE, String.class);
// Validate the business object definition search fields
validateSearchResponseFields(fieldsRequested);
// Create a new object to hold the search index response
ElasticsearchResponseDto elasticsearchResponseDto;
Set<String> facetFields = new HashSet<>();
if (CollectionUtils.isNotEmpty(searchRequest.getFacetFields())) {
facetFields.addAll(validateFacetFields(new HashSet<>(searchRequest.getFacetFields())));
}
// If the request contains search filters
if (CollectionUtils.isNotEmpty(searchRequest.getBusinessObjectDefinitionSearchFilters())) {
// Validate the search request.
validateBusinessObjectDefinitionIndexSearchRequest(searchRequest);
List<Map<SearchFilterType, List<TagEntity>>> tagEntitiesPerSearchFilter = new ArrayList<>();
// Iterate through all search filters and extract tag keys
for (BusinessObjectDefinitionSearchFilter searchFilter : searchRequest.getBusinessObjectDefinitionSearchFilters()) {
List<TagEntity> tagEntities = new ArrayList<>();
Map<SearchFilterType, List<TagEntity>> searchFilterTypeListMap = new HashMap<>();
if (BooleanUtils.isTrue(searchFilter.isIsExclusionSearchFilter())) {
searchFilterTypeListMap.put(SearchFilterType.EXCLUSION_SEARCH_FILTER, tagEntities);
validateExclusionSearchFilter(searchFilter);
} else {
searchFilterTypeListMap.put(SearchFilterType.INCLUSION_SEARCH_FILTER, tagEntities);
}
for (BusinessObjectDefinitionSearchKey searchKey : searchFilter.getBusinessObjectDefinitionSearchKeys()) {
// Get the actual tag entity from its key.
// TODO: bulk fetch tags and their children from the search index after we start indexing tags
TagEntity tagEntity = tagDaoHelper.getTagEntity(searchKey.getTagKey());
tagEntities.add(tagEntity);
// If includeTagHierarchy is true, get list of children tag entities down the hierarchy of the specified tag.
if (BooleanUtils.isTrue(searchKey.isIncludeTagHierarchy())) {
tagEntities.addAll(tagDaoHelper.getTagChildrenEntities(tagEntity));
}
}
// Collect all tag entities and their children (if included) into separate lists
tagEntitiesPerSearchFilter.add(searchFilterTypeListMap);
}
// Use the tag type entities lists to search in the search index for business object definitions
elasticsearchResponseDto = businessObjectDefinitionIndexSearchDao.searchBusinessObjectDefinitionsByTags(indexName, documentType, tagEntitiesPerSearchFilter, facetFields);
} else {
// Else get all of the business object definitions
elasticsearchResponseDto = businessObjectDefinitionIndexSearchDao.findAllBusinessObjectDefinitions(indexName, documentType, facetFields);
}
// Create a list to hold the business object definitions that will be returned as part of the search response
List<BusinessObjectDefinition> businessObjectDefinitions = new ArrayList<>();
// Retrieve all unique business object definition entities and construct a list of business object definitions based on the requested fields.
if (elasticsearchResponseDto.getBusinessObjectDefinitionIndexSearchResponseDtos() != null) {
for (BusinessObjectDefinitionIndexSearchResponseDto businessObjectDefinitionIndexSearchResponseDto : ImmutableSet.copyOf(elasticsearchResponseDto.getBusinessObjectDefinitionIndexSearchResponseDtos())) {
// Convert the business object definition entity to a business object definition and
// add it to the list of business object definitions that will be
// returned as a part of the search response
businessObjectDefinitions.add(createBusinessObjectDefinitionFromDto(businessObjectDefinitionIndexSearchResponseDto, fieldsRequested));
}
}
List<Facet> tagTypeFacets = null;
if (CollectionUtils.isNotEmpty(searchRequest.getFacetFields()) && elasticsearchResponseDto.getTagTypeIndexSearchResponseDtos() != null) {
tagTypeFacets = new ArrayList<>();
// construct a list of facet information
for (TagTypeIndexSearchResponseDto tagTypeIndexSearchResponseDto : elasticsearchResponseDto.getTagTypeIndexSearchResponseDtos()) {
List<Facet> tagFacets = new ArrayList<>();
for (TagIndexSearchResponseDto tagIndexSearchResponseDto : tagTypeIndexSearchResponseDto.getTagIndexSearchResponseDtos()) {
Facet tagFacet = new Facet(tagIndexSearchResponseDto.getTagDisplayName(), tagIndexSearchResponseDto.getCount(), FacetTypeEnum.TAG.value(), tagIndexSearchResponseDto.getTagCode(), null);
tagFacets.add(tagFacet);
}
tagTypeFacets.add(new Facet(tagTypeIndexSearchResponseDto.getDisplayName(), null, FacetTypeEnum.TAG_TYPE.value(), tagTypeIndexSearchResponseDto.getCode(), tagFacets));
}
}
// Construct business object search response.
BusinessObjectDefinitionIndexSearchResponse searchResponse = new BusinessObjectDefinitionIndexSearchResponse();
searchResponse.setBusinessObjectDefinitions(businessObjectDefinitions);
searchResponse.setFacets(tagTypeFacets);
return searchResponse;
}
Aggregations