为精简目前查询es用到的方法,以及方便收归管理和提升代码规范,现抽象出一个工具类,用于搜索和推荐业务的查询。
老代码:
List<BiggieCatBean> biggieCatList = new ArrayList<>();
// try {
// SearchResult searchResult = jestClient.execute(searchBiggie);
// if (searchResult.isSucceeded()) {
// List<SearchResult.Hit<BiggieCatBean, Void>> hits = searchResult.getHits(BiggieCatBean.class);
// hits.forEach(h -> biggieCatList.add(h.source));
// } else {
// log.error("failed to get result from es:{},request:{}", searchResult.getErrorMessage(), JSONObject.toJSONString(searchBiggie));
// }
// } catch (Exception e) {
// log.error("[zjy_119]jest execute error:", e);
// }
可以看到注释中的代码比较多,而且在多个业务场景下使用的话,会非常冗余
新代码:
List<BiggieCatBean> biggieCatList = JestSearchUtil.getSearch(jestClient, searchBiggie, BiggieCatBean.class);
使用本工具类之后,只要一行代码就可以完成相应业务
工具类具体实现
/**
* @author zoujunyuan
*/
@Slf4j
public class JestSearchUtil {
/**
* 常规的将查询内容存入集合中
* @param jestClient
* @param search 查询语句
* @param sourceType source实体类型
*/
public static <T> List<T> getSearch(JestClient jestClient, Search search, Class<T> sourceType) {
try {
SearchResult searchResult = jestClient.execute(search);
if (searchResult.isSucceeded()) {
return searchResult.getSourceAsObjectList(sourceType, true);
} else {
log.error("failed to get result from es:{},request:{}", searchResult.getErrorMessage(), JSONObject.toJSONString(search));
return Lists.newArrayList();
}
} catch (Exception e) {
log.error("jest execute failed :", e);
return Lists.newArrayList();
}
}
/**
* 将查询内容进行处理得到所需字段的集合
* @param jestClient
* @param search 查询语句
* @param sourceType source实体类型
* @param identifier 处理的方法
* @param <T>
* @param <K> 所需集合的类型
* @return
*/
public static <T, K> List<K> getSearchWithIdentify(JestClient jestClient, Search search, Class<T> sourceType, Function<T, K> identifier) {
try {
SearchResult searchResult = jestClient.execute(search);
if (searchResult.isSucceeded()) {
List<T> sourceList = searchResult.getSourceAsObjectList(sourceType, true);
return sourceList.stream().filter(Objects::nonNull).map(identifier).collect(Collectors.toList());
} else {
log.error("failed to get result from es:{},request:{}", searchResult.getErrorMessage(), JSONObject.toJSONString(search));
return Lists.newArrayList();
}
} catch (Exception e) {
log.error("jest execute failed or identify execute failed :", e);
return Lists.newArrayList();
}
}
}
Q:如何使用第二种需要处理字段的方法?
A:
// 获取查询得到的toId并转成Long类型
JestSearchUtil.getSearchWithIdentify(jestClient, search, UserRelationBean.class, (k) -> Long.valueOf(k.getToId()));
// 获取查询得到的toId
JestSearchUtil.getSearchWithIdentify(jestClient, search, UserRelationBean.class, UserRelationBean::getToId);