java Elasticsearch 查询共通方法

package ;

import com.github.pagehelper.PageInfo;
import com.google.common.collect.Lists;
import org.apache.commons.lang3.StringUtils;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.springframework.util.CollectionUtils;

import java.util.Collection;
import java.util.List;

/**
 * 搜索引擎插叙共通
 */
public class EsSearchCommon {

    /**
     * 范围查询 字段 s1 s2    参数 p1 p2 ，时间区间有交集则选中
     */
    public void queryRangeDates(BoolQueryBuilder boolQueryBuilder, String name1, String name2, String data1, String data2) {
        if (StringUtils.isNotBlank(data1) && StringUtils.isNotBlank(data2)) {
            BoolQueryBuilder vQAll = QueryBuilders.boolQuery();

            BoolQueryBuilder vQ1 = QueryBuilders.boolQuery();
            vQ1.must(QueryBuilders.rangeQuery(name1).gte(data1).lte(data2));

            BoolQueryBuilder vQ2 = QueryBuilders.boolQuery();
            vQ2.must(QueryBuilders.rangeQuery(name2).gte(data1).lte(data2));

            BoolQueryBuilder vQ3 = QueryBuilders.boolQuery();
            vQ3.must(QueryBuilders.rangeQuery(name1).lte(data1));
            vQ3.must(QueryBuilders.rangeQuery(name2).gte(data2));

            vQAll.should(vQ1);
            vQAll.should(vQ2);
            vQAll.should(vQ3);

            boolQueryBuilder.must(vQAll);
        }
    }

    /**
     * 范围查询
     */
    public void queryRange(BoolQueryBuilder boolQueryBuilder, String name, String data1, String data2) {
        if (StringUtils.isNotBlank(data1)) {
            boolQueryBuilder.must(QueryBuilders.rangeQuery(name).gte(data1));
        }
        if (StringUtils.isNotBlank(data2)) {
            boolQueryBuilder.must(QueryBuilders.rangeQuery(name).lte(data2));
        }
    }

    // 模糊查询
    public void setLikeQuery(BoolQueryBuilder boolQueryBuilder, String name, String value) {
        if (StringUtils.isNotBlank(value)) {
            boolQueryBuilder.must(QueryBuilders.wildcardQuery(name, "*" + value + "*"));
        }
    }

    // 分词查询
    public void setMatchQuery(BoolQueryBuilder boolQueryBuilder, String name, String value) {
        if (StringUtils.isNotBlank(value)) {
            boolQueryBuilder.must(QueryBuilders.matchPhraseQuery(name, value));
        }
    }

    // 精准查询
    public void setTermQuery(BoolQueryBuilder boolQueryBuilder, String name, String value) {
        if (StringUtils.isNotBlank(value)) {
            boolQueryBuilder.must(QueryBuilders.termQuery(name, value));
        }
    }

    public void setTermQueryInt(BoolQueryBuilder boolQueryBuilder, String name, Integer value) {
        if (value != null) {
            boolQueryBuilder.must(QueryBuilders.termQuery(name, value));
        }
    }

    // 精准查询 in List
    public void setTermsQuery(BoolQueryBuilder boolQueryBuilder, String name, List values) {
        if (!CollectionUtils.isEmpty(values)) {
            boolQueryBuilder.must(QueryBuilders.termsQuery(name, values));
        }
    }

    // 分词多个交集     条件1 && 条件2
    public void setMustParams(BoolQueryBuilder boolQueryBuilder, String name, String value) {
        if (StringUtils.isNotBlank(value)) {
            List<String> vs = Lists.newArrayList(value.split(","));
            BoolQueryBuilder vQueryBuilder = QueryBuilders.boolQuery();
            vs.forEach(v -> vQueryBuilder.must(QueryBuilders.matchPhraseQuery(name, v)));
            boolQueryBuilder.must(vQueryBuilder);
        }
    }

    // 分词多个合集     条件1 || 条件2
    public void setShouldParams(BoolQueryBuilder boolQueryBuilder, String name, String value) {
        if (StringUtils.isNotBlank(value)) {
            String[] groups = value.split(";");
            for (String group : groups) {
                List<String> vs = Lists.newArrayList(group.split(","));
                BoolQueryBuilder vQueryBuilder = QueryBuilders.boolQuery();
                vs.forEach(v -> vQueryBuilder.should(QueryBuilders.matchPhraseQuery(name, v)));
                boolQueryBuilder.must(vQueryBuilder);
            }
        }
    }

    public void initPageInfo(PageInfo resPage) {
        int pageNum = resPage.getPageNum();
        int prePage = pageNum;
        int nextPage = pageNum;

        // 总页数
        int pages = (int) resPage.getTotal() / resPage.getPageSize() + 1;

        // 是否有上一页
        boolean hasPreviousPage = pageNum > 1;

        // 是否有下一页
        boolean hasNextPage = pageNum < pages;

        // 上一页页号
        if (hasPreviousPage) {
            prePage = pageNum - 1;
        }

        // 下一页页号
        if (hasNextPage) {
            nextPage = pageNum + 1;
        }

        // 计算分页页号列表
        int[] navigatepageNums;
        int navigatePages = 8;
        //当总页数小于或等于导航页码数时
        if (pages <= navigatePages) {
            navigatepageNums = new int[pages];
            for (int i = 0; i < pages; i++) {
                navigatepageNums[i] = i + 1;
            }
        } else { //当总页数大于导航页码数时
            navigatepageNums = new int[navigatePages];
            int startNum = pageNum - navigatePages / 2;
            int endNum = pageNum + navigatePages / 2;

            if (startNum < 1) {
                startNum = 1;
                //(最前navigatePages页
                for (int i = 0; i < navigatePages; i++) {
                    navigatepageNums[i] = startNum++;
                }
            } else if (endNum > pages) {
                endNum = pages;
                //最后navigatePages页
                for (int i = navigatePages - 1; i >= 0; i--) {
                    navigatepageNums[i] = endNum--;
                }
            } else {
                //所有中间页
                for (int i = 0; i < navigatePages; i++) {
                    navigatepageNums[i] = startNum++;
                }
            }
        }

        resPage.setNavigatePages(navigatePages);
        resPage.setNavigatepageNums(navigatepageNums);
        resPage.setPrePage(prePage);
        resPage.setNextPage(nextPage);
        resPage.setHasPreviousPage(hasPreviousPage);
        resPage.setHasNextPage(hasNextPage);
    }
}