|
|
|
package com.yoho.search.service.scene.searchlike;
|
|
|
|
|
|
|
|
import com.alibaba.fastjson.JSONObject;
|
|
|
|
import com.yoho.search.base.utils.CollectionUtils;
|
|
|
|
import com.yoho.search.base.utils.ProductIndexEsField;
|
|
|
|
import com.yoho.search.core.es.model.SearchParam;
|
|
|
|
import com.yoho.search.dal.model.SimilarSkn;
|
|
|
|
import com.yoho.search.models.SearchApiResult;
|
|
|
|
import com.yoho.search.service.base.SearchDynamicConfigService;
|
|
|
|
import com.yoho.search.service.base.SearchRequestParams;
|
|
|
|
import com.yoho.search.service.base.index.ProductIndexBaseService;
|
|
|
|
import org.apache.commons.beanutils.BeanUtils;
|
|
|
|
import org.apache.commons.lang.StringUtils;
|
|
|
|
import org.elasticsearch.index.query.BoolQueryBuilder;
|
|
|
|
import org.elasticsearch.index.query.QueryBuilder;
|
|
|
|
import org.elasticsearch.index.query.QueryBuilders;
|
|
|
|
import org.elasticsearch.search.sort.SortBuilder;
|
|
|
|
import org.elasticsearch.search.sort.SortBuilders;
|
|
|
|
import org.elasticsearch.search.sort.SortOrder;
|
|
|
|
import org.slf4j.Logger;
|
|
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
|
|
import java.util.ArrayList;
|
|
|
|
import java.util.Arrays;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.Map;
|
|
|
|
|
|
|
|
@Service
|
|
|
|
public class BigdataSimilarSknService {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(BigdataSimilarSknService.class);
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
private SearchLikeHelper searchLikeHelper;
|
|
|
|
@Autowired
|
|
|
|
private ProductIndexBaseService productIndexBaseService;
|
|
|
|
@Autowired
|
|
|
|
private SearchDynamicConfigService searchDynamicConfigService;
|
|
|
|
|
|
|
|
public SearchApiResult searchLikeSimilarSknNotInShop(Map<String, String> paramMap) {
|
|
|
|
try {
|
|
|
|
// 1、获取参数
|
|
|
|
String productSkn = paramMap.get(SearchRequestParams.PARAM_SEARCH_PRODUCT_SKN);
|
|
|
|
if (StringUtils.isBlank(productSkn)) {
|
|
|
|
return new SearchApiResult().setCode(400).setMessage("请输入SKN");
|
|
|
|
}
|
|
|
|
// 2、检测分页参数【默认30条，最多60条】
|
|
|
|
int pageSize = searchLikeHelper.getPageSize(paramMap);
|
|
|
|
// 2、获取当前查询的SKN的基本信息
|
|
|
|
JSONObject productInfoInEs = searchLikeHelper.getProductInfoInEs(productSkn);
|
|
|
|
if (productInfoInEs == null) {
|
|
|
|
return new SearchApiResult().setCode(400).setMessage("SKN不存在");
|
|
|
|
}
|
|
|
|
//查similarskn
|
|
|
|
SimilarSkn similarSkn = new SimilarSkn();
|
|
|
|
BeanUtils.populate(similarSkn, searchLikeHelper.searchSimilarSkn(productSkn));
|
|
|
|
|
|
|
|
List<SearchParam> searchParams = new ArrayList<>();
|
|
|
|
if (StringUtils.isNotEmpty(similarSkn.getDiffShopActionSimilarSkns())) {
|
|
|
|
List<String> diffShopActionSimilarSkns = CollectionUtils.arrayToList(similarSkn.getDiffShopActionSimilarSkns().split(","));
|
|
|
|
searchParams.add(builderSimilarSknSearchParam(productInfoInEs, diffShopActionSimilarSkns));
|
|
|
|
}
|
|
|
|
if (StringUtils.isNotEmpty(similarSkn.getDiffShopActionSimilarSkns())) {
|
|
|
|
List<String> diffShopImgSimilarSkns = CollectionUtils.arrayToList(similarSkn.getDiffShopImgSimilarSkns().split(","));
|
|
|
|
searchParams.add(builderSimilarSknSearchParam(productInfoInEs, diffShopImgSimilarSkns));
|
|
|
|
}
|
|
|
|
|
|
|
|
int sameSortCount = searchDynamicConfigService.getSearchLikeNotInShopSameSortPercent() * pageSize / 100;
|
|
|
|
searchParams.add(builderSearchParam(productInfoInEs, Arrays.asList(productSkn), sameSortCount, true));
|
|
|
|
searchParams.add(builderSearchParam(productInfoInEs, Arrays.asList(productSkn), pageSize - sameSortCount, false));
|
|
|
|
|
|
|
|
// 4、获取搜索结果[截取条数]
|
|
|
|
List<Map<String, Object>> tempProductList = searchLikeHelper.queryProductList(searchParams);
|
|
|
|
if (tempProductList.size() > pageSize) {
|
|
|
|
tempProductList = CollectionUtils.safeSubList(tempProductList,0, pageSize);
|
|
|
|
}
|
|
|
|
// 5、构造真实返回结果
|
|
|
|
List<Map<String, Object>> productListResults = new ArrayList<>();
|
|
|
|
if (!tempProductList.isEmpty()) {
|
|
|
|
productListResults = productIndexBaseService.getProductListWithPricePlan(tempProductList);
|
|
|
|
}
|
|
|
|
JSONObject result = new JSONObject();
|
|
|
|
result.put("page", 1);
|
|
|
|
result.put("page_total", 1);
|
|
|
|
result.put("page_size", pageSize);
|
|
|
|
result.put("total", productListResults.size());
|
|
|
|
result.put("product_info", searchLikeHelper.genProductInfoResult(productInfoInEs));
|
|
|
|
result.put("product_list", productListResults);
|
|
|
|
// 8、结果加入缓存
|
|
|
|
return new SearchApiResult().setData(result);
|
|
|
|
} catch (Exception e) {
|
|
|
|
logger.error(e.getMessage(), e);
|
|
|
|
return new SearchApiResult().setData(null).setMessage("searchLikeNotInShop Exception").setCode(500);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private SearchParam builderSimilarSknSearchParam(JSONObject productInfoInEs, List<String> productSkns) {
|
|
|
|
// 1、设置SearchParam
|
|
|
|
SearchParam searchParam = new SearchParam();
|
|
|
|
// 2)设置query和filter
|
|
|
|
searchParam.setQuery(builderSimilarSknQueryBuilder(productInfoInEs));
|
|
|
|
searchParam.setFiter(builderSimilarSknFilterBuilder(productInfoInEs, productSkns));
|
|
|
|
// 3、设置排序规则[按打分排序]
|
|
|
|
List<SortBuilder<?>> sortBuilders = new ArrayList<>();
|
|
|
|
sortBuilders.add(SortBuilders.scoreSort().order(SortOrder.DESC));
|
|
|
|
searchParam.setSortBuilders(sortBuilders);
|
|
|
|
// 4、设置分页参数
|
|
|
|
searchParam.setOffset(0);
|
|
|
|
searchParam.setSize(productSkns.size());
|
|
|
|
// 5)设置返回的参数【节省带宽】
|
|
|
|
List<String> includeFields = productIndexBaseService.getProductIndexIncludeFields();
|
|
|
|
searchParam.setIncludeFields(includeFields);
|
|
|
|
return searchParam;
|
|
|
|
}
|
|
|
|
|
|
|
|
private SearchParam builderSearchParam(JSONObject productInfoInEs, List<String> productSkns, int pageSize, boolean inSameSort) {
|
|
|
|
// 1、设置SearchParam
|
|
|
|
SearchParam searchParam = new SearchParam();
|
|
|
|
// 2)设置query和filter
|
|
|
|
searchParam.setQuery(this.builderQueryBuilder(productInfoInEs, inSameSort ? "30%" : "20%"));
|
|
|
|
searchParam.setFiter(this.builderFilterBuilder(productInfoInEs, productSkns, inSameSort));
|
|
|
|
// 3、设置排序规则[按打分排序]
|
|
|
|
List<SortBuilder<?>> sortBuilders = new ArrayList<SortBuilder<?>>();
|
|
|
|
sortBuilders.add(SortBuilders.scoreSort().order(SortOrder.DESC));
|
|
|
|
searchParam.setSortBuilders(sortBuilders);
|
|
|
|
// 4、设置分页参数
|
|
|
|
searchParam.setOffset(0);
|
|
|
|
searchParam.setSize(pageSize);
|
|
|
|
// 5)设置返回的参数【节省带宽】
|
|
|
|
List<String> includeFields = productIndexBaseService.getProductIndexIncludeFields();
|
|
|
|
searchParam.setIncludeFields(includeFields);
|
|
|
|
return searchParam;
|
|
|
|
}
|
|
|
|
|
|
|
|
private QueryBuilder builderFilterBuilder(JSONObject productInfoInEs, List<String> notProductSkns, boolean inSameSort) {
|
|
|
|
String isGlobalInEs = productInfoInEs.getString(ProductIndexEsField.isGlobal);
|
|
|
|
boolean isGlobal = "Y".equalsIgnoreCase(isGlobalInEs);
|
|
|
|
BoolQueryBuilder boolFilter = searchLikeHelper.genDefaultSearchLikeFilter(notProductSkns, isGlobal);
|
|
|
|
// 1)设置此SKN相关的性别过滤条件
|
|
|
|
String gender = productInfoInEs.getString(ProductIndexEsField.gender);
|
|
|
|
List<String> genderList = searchLikeHelper.getGenderInfo(gender);
|
|
|
|
if (genderList != null && !genderList.isEmpty()) {
|
|
|
|
boolFilter.must(QueryBuilders.termsQuery(ProductIndexEsField.gender, genderList));
|
|
|
|
}
|
|
|
|
// 2)设置品牌或店铺信息
|
|
|
|
Integer brandId = productInfoInEs.getInteger(ProductIndexEsField.brandId);
|
|
|
|
Integer shopId = productInfoInEs.getInteger(ProductIndexEsField.shopId);
|
|
|
|
if (searchLikeHelper.isLegalInteger(shopId)) {
|
|
|
|
boolFilter.mustNot(QueryBuilders.termQuery(ProductIndexEsField.shopId, shopId));
|
|
|
|
} else if (searchLikeHelper.isLegalInteger(brandId)) {
|
|
|
|
boolFilter.mustNot(QueryBuilders.termQuery(ProductIndexEsField.brandId, brandId));
|
|
|
|
}
|
|
|
|
// 3)设置品类信息
|
|
|
|
Integer middleSortId = productInfoInEs.getInteger(ProductIndexEsField.middleSortId);
|
|
|
|
if (inSameSort) {
|
|
|
|
boolFilter.must(QueryBuilders.termQuery(ProductIndexEsField.middleSortId, middleSortId));
|
|
|
|
} else {
|
|
|
|
boolFilter.mustNot(QueryBuilders.termQuery(ProductIndexEsField.middleSortId, middleSortId));
|
|
|
|
boolFilter.mustNot(QueryBuilders.termQuery(ProductIndexEsField.smallSort_smallSort_keyword, productInfoInEs.getString(ProductIndexEsField.smallSort)));// 有些小分类同名，要排除这种情况。
|
|
|
|
}
|
|
|
|
return boolFilter;
|
|
|
|
}
|
|
|
|
|
|
|
|
private QueryBuilder builderSimilarSknFilterBuilder(JSONObject productInfoInEs, List<String> inProductSkns) {
|
|
|
|
String isGlobalInEs = productInfoInEs.getString(ProductIndexEsField.isGlobal);
|
|
|
|
boolean isGlobal = "Y".equalsIgnoreCase(isGlobalInEs);
|
|
|
|
BoolQueryBuilder boolFilter = searchLikeHelper.genDefaultSearchLikeFilter(null, isGlobal);
|
|
|
|
// 1)设置此SKN相关的性别过滤条件
|
|
|
|
String gender = productInfoInEs.getString(ProductIndexEsField.gender);
|
|
|
|
List<String> genderList = searchLikeHelper.getGenderInfo(gender);
|
|
|
|
if (genderList != null && !genderList.isEmpty()) {
|
|
|
|
boolFilter.must(QueryBuilders.termsQuery(ProductIndexEsField.gender, genderList));
|
|
|
|
}
|
|
|
|
//設置productskn
|
|
|
|
if (inProductSkns != null && !inProductSkns.isEmpty()) {
|
|
|
|
boolFilter.must(QueryBuilders.termsQuery(ProductIndexEsField.productSkn, inProductSkns));
|
|
|
|
}
|
|
|
|
// 2)设置品牌或店铺信息
|
|
|
|
Integer brandId = productInfoInEs.getInteger(ProductIndexEsField.brandId);
|
|
|
|
Integer shopId = productInfoInEs.getInteger(ProductIndexEsField.shopId);
|
|
|
|
if (searchLikeHelper.isLegalInteger(shopId)) {
|
|
|
|
boolFilter.mustNot(QueryBuilders.termQuery(ProductIndexEsField.shopId, shopId));
|
|
|
|
} else if (searchLikeHelper.isLegalInteger(brandId)) {
|
|
|
|
boolFilter.mustNot(QueryBuilders.termQuery(ProductIndexEsField.brandId, brandId));
|
|
|
|
}
|
|
|
|
return boolFilter;
|
|
|
|
}
|
|
|
|
|
|
|
|
private QueryBuilder builderQueryBuilder(JSONObject productInfoInEs, String minimumShouldMatch) {
|
|
|
|
String queryString = searchLikeHelper.genYohoQueryStringWithOutBrandName(productInfoInEs);
|
|
|
|
String productFeatureFactor = productInfoInEs.getString(ProductIndexEsField.productFeatureFactor);
|
|
|
|
QueryBuilder queryBuilder = searchLikeHelper.genSearchLikeQueryBuilder(queryString, minimumShouldMatch, productFeatureFactor);
|
|
|
|
return queryBuilder;
|
|
|
|
}
|
|
|
|
|
|
|
|
private QueryBuilder builderSimilarSknQueryBuilder(JSONObject productInfoInEs) {
|
|
|
|
return searchLikeHelper.genProductFeatureQueryBuilder( productInfoInEs.getString(ProductIndexEsField.productFeatureFactor));
|
|
|
|
}
|
|
|
|
|
|
|
|
} |
...
|
...
|
|
