ES学习笔记（2）--- javaAPI搜索篇

先来几张大神同事分享的ppt

多的不说，快快上码

public PageInfo<SearchResult> getAllRelatedPage(String keyword, String sortRule ,int pageSize,int pageNum) throws PcompException {        try {            String[] indexs = {Constants.PUBLIC_COMPONENT_SYSTEM, Constants.INNO_WORKS_SYSTEM};            String[] types = {PcompConstants.PCOMP_SOFTWARE, PcompConstants.PCOMP_SOFTWARE_ARTICLE, PcompConstants.PCOMP_VERSION, InnovationConstants.INNOVATION_WORKS};            SearchHits searchHits = getSearchHitsPage(keyword,indexs, types, sortRule, pageSize, pageNum);            if (searchHits == null) {                return null;            }            List<SearchResult> searchResults = getSearchResults(searchHits.getHits());            return getResultPageInfo(searchResults, pageSize, pageNum, searchHits);        } catch (Exception e) {            throw new PcompException(e);        }    }

private SearchHits getSearchHitsPage(String keyword, String[] indexs ,String[] types, String sortRule,  int pageSize , int pageNum) {        String[] stringMeta = esClient.getAllStringMeta(indexs, types);        if (stringMeta == null) {            return null;        }        MultiMatchQueryBuilder termQueryBuilder = QueryBuilders.multiMatchQuery(keyword, stringMeta);        QueryBuilder qb = new BoolQueryBuilder()                .must(QueryBuilders.termQuery("isDeleted", false))                .must(termQueryBuilder);        return esClient.searchDocumentPageSort(indexs, types, qb, sortRule, pageSize, pageNum);    }

/**     * 获取要搜索的目标域（例如：要搜索content、name等字段，就会返回["content","name"]）     * @param indexs     * @param types     * @return     */    public String[] getAllStringMeta(String indexs[], String types[]) {        HashSet<String> keySets = new HashSet<>();        String[] tempTypes = types;        HashMap<String,List<String>> hashMap = new HashMap<String,List<String>>();        List<String> list = new ArrayList<>();        list.add(PcompConstants.PCOMP_SOFTWARE);        list.add(PcompConstants.PCOMP_SOFTWARE_ARTICLE);        list.add(PcompConstants.PCOMP_VERSION);        hashMap.put(Constants.PUBLIC_COMPONENT_SYSTEM,list);        List<String> list1 = new ArrayList<>();        list1.add(InnovationConstants.INNOVATION_WORKS);        hashMap.put(Constants.INNO_WORKS_SYSTEM,list1);        try {            for(String index : indexs){                GetMappingsResponse getMappingsResponse = client.admin().indices().prepareGetMappings(index).get();                List<String> tempList = hashMap.get(index);                for(String type : tempList){                    MappingMetaData mapMeta =  getMappingsResponse.getMappings().get(index).get(type);                    if(mapMeta == null){                        break;                    }                    HashMap properties =  (HashMap)mapMeta.getSourceAsMap().get("properties");                    for (Object object : properties.keySet()) {                        if (((HashMap) properties.get(object)).get("type").equals("string")) {                            keySets.add((String) object);                        }                    }                }            }            return keySets.toArray(new String[]{});        } catch (Exception e) {            log.warn("Caught:", e);            return null;        }    }

//自带排序效果    public SearchHits searchDocumentPageSort(String[] indexes, String[] types, QueryBuilder queryBuilder , String sortRule, int pageSize , int pageNum) {        SortBuilder sortBuilder = new ScoreSortBuilder();        SearchRequestBuilder searchRequestBuilder = client.prepareSearch(indexes);        switch (sortRule) {        //相关度倒排        case SortRule.BY_CORRELATION_DESC:            sortBuilder.order(SortOrder.DESC);            searchRequestBuilder = searchRequestBuilder.addSort(sortBuilder);            break;        //相关度正排        case SortRule.BY_CORRELATION:            sortBuilder.order(SortOrder.ASC);            searchRequestBuilder = searchRequestBuilder.addSort(sortBuilder);            break;        //时间到排        case SortRule.BY_MODIFIED_TIME_DESC:            searchRequestBuilder = searchRequestBuilder.addSort("modifiedTime" , SortOrder.DESC);            break;        //时间正排        case SortRule.BY_MODIFIED_TIME:            searchRequestBuilder = searchRequestBuilder.addSort("modifiedTime" , SortOrder.ASC);            break;        }        SearchResponse searchResponse = searchRequestBuilder                .setTypes(types)                .addHighlightedField("*")                .setHighlighterRequireFieldMatch(false)                .setQuery(queryBuilder)                .setHighlighterPreTags("<strong>")                .setHighlighterPostTags("</strong>")                .setSize(pageSize)                .setFrom((pageNum - 1) * pageSize)                .execute()                .actionGet();        return searchResponse.getHits();    }

/**     * 将搜索结果转为List<SearchResult>     * @param searchHit     * @return     * @throws PcompException     */    private List<SearchResult> getSearchResults(SearchHit[] searchHit)throws PcompException{        List<SearchResult> searchResults = new LinkedList<>();        for (int i = 0; i < searchHit.length; i++) {            String json = searchHit[i].getSourceAsString();            SearchResult searchResult = new SearchResult();            switch (searchHit[i].type()){            case PcompConstants.PCOMP_SOFTWARE :                searchResult = getResult(JSON.parseObject(json, PcompSoftware.class),searchHit[i].getHighlightFields());break;            case PcompConstants.PCOMP_SOFTWARE_ARTICLE :                searchResult = getResult(JSON.parseObject(json, PcompSoftwareArticle.class),searchHit[i].getHighlightFields());break;            case PcompConstants.PCOMP_VERSION :                searchResult = getResult(JSON.parseObject(json, PcompVersion.class),searchHit[i].getHighlightFields());break;            case InnovationConstants.INNOVATION_WORKS :                searchResult = getResult(JSON.parseObject(json, Innovation.class),searchHit[i].getHighlightFields());break;            }            searchResults.add(searchResult);        }        return searchResults;

/**     * 将结果对象（pcompSoftware、pcompVersion等等）转为SearchResult对象。     * @param object     * @return     * @throws PcompException     */    private SearchResult getResult(Object object, Map<String,HighlightField> highlightFieldMap) throws PcompException {        SearchResult searchResult = new SearchResult();        SimpleDateFormat simpleDateFormat = simpleDateFormatThreadLocal.get();        if (object instanceof PcompSoftware) {            PcompSoftware pcompSoftware = (PcompSoftware) object;            searchResult.setId(pcompSoftware.getId());            searchResult.setTitle(pcompSoftware.getName());            searchResult.setLink(StrUtils.makeString("opensoftware.html?pcompSoftwareId=", pcompSoftware.getId()));//            searchResult.setIntroduction(IntroductionUtils.getShortIntroduction(pcompSoftware.getIntroduction()));            searchResult.setIntroduction(highlightFieldMap.get("introduction")==null?pcompSoftware.getIntroduction():highlightFieldMap.get("introduction").toString());            searchResult.setAvatar(pcompSoftware.getAvatar());            searchResult.setModifiedTime(simpleDateFormat.format(pcompSoftware.getModifiedTime()));            searchResult.setPath(StrUtils.makeString("opensoftware.html?pcompSoftwareId=", pcompSoftware.getId()));        } else if (object instanceof PcompVersion) {            PcompVersion pcompVersion = (PcompVersion) object;            searchResult.setId(pcompVersion.getId());            PcompSoftware pcompSoftware = pcompSoftwareManager.getPcompSoftwareByPcompSoftwareId(pcompVersion.getPcompSoftwareId());            searchResult.setTitle(StrUtils.makeString(                    pcompSoftware.getName(),                    Constants.NAME_SEPARATOR, pcompVersion.getVersionNumber()));//            searchResult.setIntroduction(IntroductionUtils.getShortIntroduction(pcompVersion.getIntroduction()));            searchResult.setIntroduction(highlightFieldMap.get("introduction")==null?pcompVersion.getIntroduction():highlightFieldMap.get("introduction").toString());            searchResult.setModifiedTime(simpleDateFormat.format(pcompVersion.getModifiedTime()));            searchResult.setAvatar(pcompSoftware.getAvatar());                searchResult.setLink(StrUtils.makeString("opensoftware.html?pcompSoftwareId=",pcompSoftware.getId(),"&category=history&versionId=",pcompVersion.getId()));            searchResult.setPath(StrUtils.makeString("opensoftware.html?pcompSoftwareId=",pcompSoftware.getId(),"&pcompVersionId=",pcompVersion.getId()));        }else if (object instanceof PcompSoftwareArticle) {            PcompSoftwareArticle pcompSoftwareArticle = (PcompSoftwareArticle) object;            searchResult.setId(pcompSoftwareArticle.getId());            PcompSoftware pcompSoftware = pcompSoftwareManager.getPcompSoftwareByPcompSoftwareId(pcompSoftwareArticle.getPcompSoftwareId());            searchResult.setTitle(StrUtils.makeString(pcompSoftware.getName(),                    Constants.NAME_SEPARATOR,                    pcompSoftwareArticle.getTitle()));//            searchResult.setIntroduction(IntroductionUtils.getShortIntroduction(pcompSoftwareArticle.getContent()));            searchResult.setIntroduction(highlightFieldMap.get("content")==null?pcompSoftwareArticle.getContent():highlightFieldMap.get("content").toString());            searchResult.setModifiedTime(simpleDateFormat.format(pcompSoftwareArticle.getModifiedTime()));            searchResult.setAvatar(pcompSoftware.getAvatar());            searchResult.setLink(StrUtils.makeString("opensoftware.html?pcompSoftwareId=",pcompSoftware.getId(),"&category=article&articleId=",pcompSoftwareArticle.getId()));            searchResult.setPath(StrUtils.makeString("opensoftware.html?pcompSoftwareId=",pcompSoftware.getId(),"&pcompArticleId=",pcompSoftwareArticle.getId()));        }else if (object instanceof AcademyActivity) {            AcademyActivity academyActivity = (AcademyActivity) object;            searchResult.setId(academyActivity.getId());            searchResult.setTitle(StrUtils.makeString(academyActivity.getTitle()));            searchResult.setIntroduction(IntroductionUtils.getShortIntroduction(academyActivity.getContent()));            searchResult.setModifiedTime(simpleDateFormat.format(academyActivity.getModifiedTime()));            searchResult.setAvatar(academyActivity.getImage());            searchResult.setLink(StrUtils.makeString("academyActivity.html?academyActivityId=",academyActivity.getId()));            searchResult.setPath(StrUtils.makeString("academyActivity.html?academyActivityId=", academyActivity.getId()));        }        else if (object instanceof AcademyDepartment) {            AcademyDepartment academyDepartment = (AcademyDepartment) object;            searchResult.setId(academyDepartment.getId());            searchResult.setTitle(StrUtils.makeString(academyDepartment.getName()));            searchResult.setIntroduction(IntroductionUtils.getShortIntroduction(academyDepartment.getIntroduction()));            searchResult.setModifiedTime(simpleDateFormat.format(academyDepartment.getModifiedTime()));            searchResult.setAvatar(academyDepartment.getAvatar());            searchResult.setLink(StrUtils.makeString("academyDepartment.html?academyDepartmentId=", academyDepartment.getId()));            searchResult.setPath(                    StrUtils.makeString("academyDepartment.html?academyDepartmentId=", academyDepartment.getId()));        }else if (object instanceof Innovation) {            Innovation innovation = (Innovation) object;            searchResult.setId(innovation.getId().toString());            searchResult.setTitle(innovation.getWorksName());            searchResult.setIntroduction(highlightFieldMap.get("content")==null?innovation.getContent():highlightFieldMap.get("content").toString());            searchResult.setModifiedTime(simpleDateFormat.format(innovation.getModifiedTime()));            searchResult.setAvatar(innovation.getWorksImage());            searchResult.setPath(StrUtils.makeString("innodetail.html?workId=" , innovation.getId()));            searchResult.setLink(StrUtils.makeString("innodetail.html?workId=" , innovation.getId()));        } else {            searchResult = null;        }        return searchResult;    }

 /**     * 将List<SearchResult> searchResults转为PageInfo<SearchResult>，     * 即为PageInfo<SearchResult>赋上页面信息，页面展示的是PageInfo<SearchResult>对象     * @param searchResults     * @param pageSize     * @param pageNum     * @param searchHit     * @return     */    private PageInfo<SearchResult> getResultPageInfo(List<SearchResult> searchResults, int pageSize, int pageNum, SearchHits searchHit){        PageInfo<SearchResult> searchPageInfo = new PageInfo<>();        searchPageInfo.setList(searchResults);        searchPageInfo.setPageSize(pageSize);        searchPageInfo.setFirstPage(1);        int lastPage = (int) (searchHit.getTotalHits() % pageSize == 0 ?                searchHit.getTotalHits() / pageSize :                searchHit.getTotalHits() / pageSize + 1);        searchPageInfo.setLastPage(lastPage);        searchPageInfo.setIsFirstPage(pageNum == 1);        searchPageInfo.setIsLastPage(pageNum == lastPage);        searchPageInfo.setHasPreviousPage(!searchPageInfo.isIsFirstPage());        searchPageInfo.setHasNextPage(!searchPageInfo.isIsLastPage());        searchPageInfo.setPageNum(pageNum);        int [] pageNums =  new int[lastPage];        for(int i = 0; i < lastPage; i++){            pageNums[i] = i+1;        }        searchPageInfo.setNavigatepageNums(pageNums);        searchPageInfo.setTotal(searchHit.getTotalHits());        return searchPageInfo;    }