上节讲完普通索引的创建以及索引文档的插入、修改、删除和文档分页查询。
接着讲 competion suggester 的索引。
competion suggester 提供自动完成 / 搜索即用类型功能。
是一个导航功能,以引导用户在键入相关结果时,提高搜索精度。
它不是用来拼写纠正或确实是你的意思功能,如术语或短语暗示。
理想情况下,自动完成功能应该和用户类型一样快,以提供与用户已经输入的内容相关的即时反馈。
因此,competion suggester 是优化速度。
suggester 使用的数据结构能够快速查找,但是构建和存储在内存中的成本很高。
创建索引
/***
* <p>
* 创建索引:
* <p>
* @param indexName 索引名称
* @param completionIndexFieldEntityList 索引包含的字段
*/
public ResponseResult createCompletionIndex(String indexName, List<CompletionIndexFieldEntity> completionIndexFieldEntityList) {
IdxVo idxVo = new IdxVo();
idxVo.setIdxName(indexName);
IdxVo.IdxSql tidxSql = new IdxVo.IdxSql();
tidxSql.setDynamic(false);
for (CompletionIndexFieldEntity indexFieldEntity : completionIndexFieldEntityList) {
Map<String, Object> map = new HashMap();
map.put(ESConstants.IndexConstans.TYPE, indexFieldEntity.getType());
// completion 的专用
if (StringUtils.isNotEmpty(indexFieldEntity.getPreservePositionIncrements())) {
map.put(ESConstants.IndexConstans.PRESERVE_POSITION_INCREMENTS, indexFieldEntity.getPreservePositionIncrements());
}
if (StringUtils.isNotEmpty(indexFieldEntity.getPreserveSeparators())) {
map.put(ESConstants.IndexConstans.PRESERVE_SEPARATORS, indexFieldEntity.getPreserveSeparators());
}
if (indexFieldEntity.getMaxInputLength() > -1) {
map.put(ESConstants.IndexConstans.MAX_INPUT_LENGTH, indexFieldEntity.getMaxInputLength());
}
if (indexFieldEntity.getContextsMapList() != null && indexFieldEntity.getContextsMapList().size() > 0) {
map.put(ESConstants.IndexConstans.CONTEXTS, indexFieldEntity.getContextsMapList());
}
tidxSql.getProperties().put(indexFieldEntity.getName(), map);
}
idxVo.setIdxSql(tidxSql);
ResponseResult response = createIndex(idxVo);
return response;
}
插入文档
这里插入几个测试文档,包含上下文 category_code 和 channel
public void insertIndex4CompletionWithContexts() {
Map<String, Object> suggestMap = new HashMap<>();
String id = System.currentTimeMillis() + "_myId";
List<String> inputList = new ArrayList<>();
inputList.add("头发长,见识短");
inputList.add("见识短,头发长");
inputList.add(id);
Map<String, List<String>> contextsMap1 = new HashMap<>();
List<String> categoryCodeList = new ArrayList<>();
categoryCodeList.add("100003");
categoryCodeList.add("100001");
contextsMap1.put("category_code", categoryCodeList);
List<String> channelList = new ArrayList<>();
channelList.add("web");
channelList.add("miniapp");
contextsMap1.put("channel", channelList);
suggestMap.put("input", inputList);
suggestMap.put("contexts", contextsMap1);
//suggestMap.put("id", id);
Map<String, Map> rootMap = new HashMap<>();
rootMap.put("suggest", suggestMap);
ElasticEntity entity = new ElasticEntity();
entity.setId(id);
entity.setData(rootMap);
elasticService.insertOrUpdateOne(indexName, entity);
log.debug("batch insert success");
}
/**
* @param idxName index
* @param entity 对象
* @return void
*/
public void insertOrUpdateOne(String idxName, ElasticEntity entity) {
IndexRequest request = new IndexRequest(idxName);
log.info("Data : id={},entity={}", entity.getId(), JSON.toJSONString(entity.getData()));
request.id(entity.getId());
request.source(JSON.toJSONString(entity.getData()), XContentType.JSON);
try {
restHighLevelClient.index(request, RequestOptions.DEFAULT);
} catch (Exception e) {
throw new RuntimeException(e);
}
}
查询
查询的时候,可以指定上下文进行范围圈定。
@Test
public void testCompletionSuggestion(){
String completionFieldName = "suggest";
String prefixWord = "佛山";
Map<String, List> contextMap = new HashMap<>();
List<String> categoryCodeList = new ArrayList<>();
categoryCodeList.add("100001");
categoryCodeList.add("100003");
contextMap.put("category_code", categoryCodeList);
List<String> channelList = new ArrayList<>();
channelList.add("web");
channelList.add("h5");
contextMap.put("channel", channelList);
List<String> completionSuggestList = elasticService.getCompletionSuggestList(indexName, completionFieldName, contextMap, prefixWord);
log.error("---testSuggestion over!--" + JSON.toJSONString(completionSuggestList));
}