spring boot 集成 elasticsearch 7.x (四）

深碍√TFBOYSˉ_ 2023-03-12 09:18 20阅读 0赞

上节讲完普通索引的创建以及索引文档的插入、修改、删除和文档分页查询。  
接着讲 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));
    
        }