github/mall
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

elasticsearch 聚合搜索结果集映射修改

Browse Source
This commit is contained in:
macro 2019-06-08 10:32:08 +08:00
parent df5a6b28f4
commit e0bcd0ced4
1 changed files with 20 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
mall-search/src/main/java/com/macro/mall/search/service/impl/EsProductServiceImpl.java
View File

@ -15,6 +15,7 @@ import org.elasticsearch.search.aggregations.AbstractAggregationBuilder;
import org.elasticsearch.search.aggregations.Aggregation;
import org.elasticsearch.search.aggregations.AggregationBuilders;
import org.elasticsearch.search.aggregations.bucket.filter.InternalFilter;
import org.elasticsearch.search.aggregations.bucket.nested.InternalNested;
import org.elasticsearch.search.aggregations.bucket.terms.LongTerms;
import org.elasticsearch.search.aggregations.bucket.terms.StringTerms;
import org.elasticsearch.search.aggregations.bucket.terms.Terms;
@ -247,25 +248,25 @@ public class EsProductServiceImpl implements EsProductService {
productRelatedInfo.setProductCategoryNames(productCategoryNameList);
//设置参数
Aggregation productAttrs = aggregationMap.get("allAttrValues");
// List<Terms.Bucket> attrIds = ((LongTerms) ((InternalFilter)productAttrs.getProperty("productAttrs")).getAggregations().getProperty("attrIds")).getBuckets();
// List<EsProductRelatedInfo.ProductAttr> attrList = new ArrayList<>();
// for (Terms.Bucket attrId : attrIds) {
// EsProductRelatedInfo.ProductAttr attr = new EsProductRelatedInfo.ProductAttr();
// attr.setAttrId((Long) attrId.getKey());
// List<String> attrValueList = new ArrayList<>();
// List<Terms.Bucket> attrValues = ((StringTerms) attrId.getAggregations().get("attrValues")).getBuckets();
// List<Terms.Bucket> attrNames = ((StringTerms) attrId.getAggregations().get("attrNames")).getBuckets();
// for (Terms.Bucket attrValue : attrValues) {
// attrValueList.add(attrValue.getKeyAsString());
// }
// attr.setAttrValues(attrValueList);
// if(!CollectionUtils.isEmpty(attrNames)){
// String attrName = attrNames.get(0).getKeyAsString();
// attr.setAttrName(attrName);
// }
// attrList.add(attr);
// }
// productRelatedInfo.setProductAttrs(attrList);
List<LongTerms.Bucket> attrIds = ((LongTerms) ((InternalFilter) ((InternalNested) productAttrs).getProperty("productAttrs")).getProperty("attrIds")).getBuckets();
List<EsProductRelatedInfo.ProductAttr> attrList = new ArrayList<>();
for (Terms.Bucket attrId : attrIds) {
EsProductRelatedInfo.ProductAttr attr = new EsProductRelatedInfo.ProductAttr();
attr.setAttrId((Long) attrId.getKey());
List<String> attrValueList = new ArrayList<>();
List<StringTerms.Bucket> attrValues = ((StringTerms) attrId.getAggregations().get("attrValues")).getBuckets();
List<StringTerms.Bucket> attrNames = ((StringTerms) attrId.getAggregations().get("attrNames")).getBuckets();
for (Terms.Bucket attrValue : attrValues) {
attrValueList.add(attrValue.getKeyAsString());
}
attr.setAttrValues(attrValueList);
if(!CollectionUtils.isEmpty(attrNames)){
String attrName = attrNames.get(0).getKeyAsString();
attr.setAttrName(attrName);
}
attrList.add(attr);
}
productRelatedInfo.setProductAttrs(attrList);
return productRelatedInfo;
}
}