拉取代码
git拉取 yxh-elasticsearch: es基本封装工具
使用场景
我们拿游览器举例,我将我要搜索的内容输入到输入框进行搜索,游览器就会根据对应的内容查出文章中出现过的关键字,并加上样式,让我们看的更清晰。
我们以就是使用全文检索的时候,好比用户输入关键字,es进行查询,给用户呈现的页面对应的内容,以及将用户输入的关键字,加粗或标识出来。
获取泛型
/**
* 获取泛型
*/
private Class<T> deSerializable() {
Type type = getClass().getGenericSuperclass();
if (type instanceof ParameterizedType) {
ParameterizedType parameterizedType = (ParameterizedType) type;
System.out.println(parameterizedType.getActualTypeArguments()[0]);
return (Class<T>) parameterizedType.getActualTypeArguments()[0];
}
throw new RuntimeException();
}
接口
接口需要加上泛型
核心代码
高亮查询
这块代码主要就是实现一个查询出高亮字段,通过读取注解里的参数,做对应的替换。
/**
* 高亮查询
* @param request 参数
* @param data 泛型
* @param indexName 索引名
* @param client 连接
* */
public List<Object> queryHighLight(String request, Class<?> data, String indexName, RestHighLevelClient client) throws IllegalAccessException {
List<Object> list = new ArrayList<>();
//获取list的泛型
HighLightDTO highLightDTO = readHighLight(data);
try {
//创建请求索引
SearchRequest searchRequest = new SearchRequest(indexName);
//创建构造器
SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
//创建条件拼接
BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
//查询用户名 模糊查询
if (null != request){
boolQueryBuilder.must(QueryBuilders.wildcardQuery(highLightDTO.getName(),"*"+request+"*"));
}
//条件存放构造器
searchSourceBuilder.query(boolQueryBuilder);
//高亮
searchSourceBuilder.highlighter(new HighlightBuilder().field(highLightDTO.getName()).preTags(highLightDTO.getPreTag()).postTags(highLightDTO.getPostTag()));
searchRequest.source(searchSourceBuilder);
SearchResponse search = client.search(searchRequest, RequestOptions.DEFAULT);
SearchHits hits = search.getHits();
for (SearchHit hit : hits.getHits()) {
String sourceAsString = hit.getSourceAsString();
Object object = JSONObject.parseObject(sourceAsString, data);
//获取高亮集合
Map<String, HighlightField> highlightFields = hit.getHighlightFields();
if (null!=highlightFields){
HighlightField obj = highlightFields.get(highLightDTO.getName());
if (null != obj){
String str = "";
for (Text fragment : obj.getFragments()) {
str += fragment;
}
//将字段赋值
assignment(object, highLightDTO.getMapping(), str);
}
}
//将存储的结果存入list集合
list.add(object);
}
} catch (Exception e) {
throw new RuntimeException(e);
}
return list;
}
标记映射
主要起到,读取@HighLight该注解所映射的字段。
/**
* 读取@HighLight映射字段
* */
private HighLightDTO readHighLight(Class<?> data) throws IllegalAccessException {
Field[] fields = data.getDeclaredFields();
String name = null;
String preTags = null;
String postTags = null;
String mapping = null;
for (Field field : fields) {
HighLight annotation = field.getAnnotation(HighLight.class);//映射@highLight值
if (annotation!=null){
field.setAccessible(true);
name = field.getName();//字段名
preTags = annotation.preTag();//自定义标签前缀
postTags = annotation.postTag();//自定义标签后缀
mapping = annotation.mappingField();//映射字段
}
}
return new HighLightDTO(name,preTags,postTags,mapping);
}
/**
* 给高亮值赋值
*/
private void assignment(Object data, String fieldName, String newValue) throws IllegalAccessException, NoSuchFieldException {
Field field = data.getClass().getDeclaredField(fieldName);
field.setAccessible(true);
field.set(data, convertValue(field.getType(), newValue));
}
/**
* 转类型
*/
private Object convertValue(Class<?> fieldType, String value) {
JdkDataTypeEnum byType = JdkDataTypeEnum.getByType(String.valueOf(fieldType));
switch (byType){
case INTEGER:
return Integer.parseInt(value);
case LONG:
return Long.parseLong(value);
case BOOLEAN:
return Boolean.parseBoolean(value);
default:
return value;
}
}
总结
这次我是将原先的yxh-es做了一个迭代,一开始没有查询方法,也没有高亮,当时这个高亮灵感是来自于easy-es,因为之前在使用这个工具的时候,发现他们团的做的少了个高亮查询,我测了好几次,让身边的同事也测了,发现不行,就是有问题的,想着就自己封装es,把这个高亮也补上。当然我这个高亮目前仅限于一个字段来模糊查,我后门会在对该工具迭代更新。