ElasticSearch 条件,分页,排序,多条件查询Demo

原文:

java1234,峰哥

1. 获取客户端连接demo

import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.transport.InetSocketTransportAddress;
import org.elasticsearch.transport.client.PreBuiltTransportClient;
import org.junit.After;
import org.junit.Before;

import java.net.InetAddress;
import java.net.UnknownHostException;

/**
 * @author XXL
 * @create 2019-08-11 11:00
 */
public class ClientDemo {

    protected TransportClient client = null;

    @Before
    public void setUp() throws UnknownHostException {
        Settings settings = Settings.builder()
                .put("cluster.name", "my-application")
                .build();

        client = new PreBuiltTransportClient(settings)
                .addTransportAddress(new InetSocketTransportAddress(InetAddress.getByName("公网ip"),9300));

        System.out.println("链接成功");
    }

    @After
    public void close(){
        if(client != null){
            client.close();
        }
    }

}

2. demo示例:

import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import org.elasticsearch.action.index.IndexResponse;
import org.elasticsearch.action.search.SearchRequestBuilder;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.common.xcontent.XContentType;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
import org.elasticsearch.search.fetch.subphase.highlight.HighlightBuilder;
import org.elasticsearch.search.sort.SortOrder;
import org.junit.Test;

/**
 * @author XXL
 * @create 2019-08-11 11:08
 */
public class SearchDemo extends ClientDemo {

    /**
     * 添加文档
     */
    @Test
    public void testIndex(){
        JsonArray jsonArray = new JsonArray();

        JsonObject jsonObject=new JsonObject();
        jsonObject.addProperty("title", "前任3:再见前任");
        jsonObject.addProperty("publishDate", "2017-12-29");
        jsonObject.addProperty("content", "一对好基友孟云(韩庚 饰)和余飞(郑恺 饰)跟女友都因为一点小事宣告分手,并且“拒绝挽回,死不认错”。两人在夜店、派对与交友软件上放飞人生第二春,大肆庆祝“黄金单身期”,从而引发了一系列好笑的故事。孟云与女友同甘共苦却难逃“五年之痒”,余飞与女友则棋逢敌手相爱相杀无绝期。然而现实的“打脸”却来得猝不及防:一对推拉纠结零往来,一对纠缠互怼全交代。两对恋人都将面对最终的选择:是再次相见?还是再也不见?");
        jsonObject.addProperty("director", "田羽生");
        jsonObject.addProperty("price", "35");
        jsonArray.add(jsonObject);

        JsonObject jsonObject2=new JsonObject();
        jsonObject2.addProperty("title", "机器之血");
        jsonObject2.addProperty("publishDate", "2017-12-29");
        jsonObject2.addProperty("content", "2007年,Dr.James在半岛军火商的支持下研究生化人。研究过程中,生化人安德烈发生基因突变大开杀戒,将半岛军火商杀害,并控制其组织,接管生化人的研究。Dr.James侥幸逃生,只好寻求警方的保护。特工林东(成龙 饰)不得以离开生命垂危的小女儿西西,接受证人保护任务...十三年后,一本科幻小说《机器之血》的出版引出了黑衣生化人组织,神秘骇客李森(罗志祥 饰)(被杀害的半岛军火商的儿子),以及隐姓埋名的林东,三股力量都开始接近一个“普通”女孩Nancy(欧阳娜娜 饰)的生活,想要得到她身上的秘密。而黑衣人幕后受伤隐藏多年的安德烈也再次出手,在多次缠斗之后终于抓走Nancy。林东和李森,不得不以身犯险一同前去解救,关键时刻却发现李森竟然是被杀害的半岛军火商的儿子,生化人的实验记录也落入了李森之手......");
        jsonObject2.addProperty("director", "张立嘉");
        jsonObject2.addProperty("price", "45");
        jsonArray.add(jsonObject2);

        JsonObject jsonObject3=new JsonObject();
        jsonObject3.addProperty("title", "星球大战8:最后的绝地武士");
        jsonObject3.addProperty("publishDate", "2018-01-05");
        jsonObject3.addProperty("content", "《星球大战:最后的绝地武士》承接前作《星球大战:原力觉醒》的剧情,讲述第一军团全面侵袭之下,蕾伊(黛西·雷德利 Daisy Ridley 饰)、芬恩(约翰·博耶加 John Boyega 饰)、波·达默龙(奥斯卡·伊萨克 Oscar Isaac 饰)三位年轻主角各自的抉 择和冒险故事。前作中觉醒强大原力的蕾伊独自寻访隐居的绝地大师卢克·天行者(马克·哈米尔 Mark Hamill 饰),在后者的指导下接受原力训练。芬恩接受了一项几乎不可能完成的任务,为此他不得不勇闯敌营,面对自己的过去。波·达默龙则要适应从战士向领袖的角色转换,这一过程中他也将接受一些血的教训。");
        jsonObject3.addProperty("director", "莱恩·约翰逊");
        jsonObject3.addProperty("price", "55");
        jsonArray.add(jsonObject3);

        JsonObject jsonObject4=new JsonObject();
        jsonObject4.addProperty("title", "羞羞的铁拳");
        jsonObject4.addProperty("publishDate", "2017-12-29");
        jsonObject4.addProperty("content", "靠打假拳混日子的艾迪生(艾伦 饰),本来和正义感十足的体育记者马小(马丽 饰)是一对冤家,没想到因为一场意外的电击,男女身体互换。性别错乱后,两人互坑互害,引发了拳坛的大地震,也揭开了假拳界的秘密,惹来一堆麻烦,最终两人在“卷莲门”副掌门张茱萸(沈腾 饰)的指点下,向恶势力挥起了羞羞的铁拳。");
        jsonObject4.addProperty("director", "宋阳 / 张吃鱼");
        jsonObject4.addProperty("price", "35");
        jsonArray.add(jsonObject4);

        JsonObject jsonObject5=new JsonObject();
        jsonObject5.addProperty("title", "战狼2");
        jsonObject5.addProperty("publishDate", "2017-07-27");
        jsonObject5.addProperty("content", "故事发生在非洲附近的大海上,主人公冷锋(吴京 饰)遭遇人生滑铁卢,被“开除军籍”,本想漂泊一生的他,正当他打算这么做的时候,一场突如其来的意外打破了他的计划,突然被卷入了一场非洲国家叛乱,本可以安全撤离,却因无法忘记曾经为军人的使命,孤身犯险冲回沦陷区,带领身陷屠杀中的同胞和难民,展开生死逃亡。随着斗争的持续,体内的狼性逐渐复苏,最终孤身闯入战乱区域,为同胞而战斗。");
        jsonObject5.addProperty("director", "吴京");
        jsonObject5.addProperty("price", "38");
        jsonArray.add(jsonObject5);

        for(int i = 0; i < jsonArray.size(); i ++){
            JsonObject jo = jsonArray.get(i).getAsJsonObject();
            IndexResponse response = client.prepareIndex("firm", "dongzuo")
                    .setSource(jo.toString(), XContentType.JSON).get();
            System.out.println("索引名称:"+response.getIndex());
            System.out.println("类型:"+response.getType());
            System.out.println("文档ID:"+response.getId());
            System.out.println("当前实例状态:"+response.status());
        }
    }

    /**
     * 查询所有
     */
    @Test
    public void getAll(){
        SearchResponse searchResponse = client.prepareSearch("firm").setTypes("dongzuo")
                .setQuery(QueryBuilders.matchAllQuery()).execute().actionGet();
        SearchHits hits = searchResponse.getHits();
        for(SearchHit hit : hits){
            System.out.println(hit.getSourceAsString());
        }
    }

    /**
     * 分页查询
     */
    @Test
    public void page(){
        SearchRequestBuilder srb = client.prepareSearch("firm").setTypes("dongzuo");
        SearchResponse response = srb.setQuery(QueryBuilders.matchAllQuery())
                .setFrom(0)
                .setSize(2)
                .execute().actionGet();
        SearchHits hits = response.getHits();
        for(SearchHit hit : hits){
            System.out.println(hit.getSourceAsString());
        }
    }

    /**
     * 排序查询
     */
    @Test
    public void sort(){
        SearchRequestBuilder srb = client.prepareSearch("firm").setTypes("dongzuo");
        SearchResponse response = srb.setQuery(QueryBuilders.matchAllQuery())
                .addSort("publishDate", SortOrder.DESC)
                .execute().actionGet();
        SearchHits hits = response.getHits();
        for(SearchHit hit : hits){
            System.out.println(hit.getSourceAsString());
        }
    }

    /**
     * 数据过滤
     */
    @Test
    public void include(){
        SearchRequestBuilder srb = client.prepareSearch("firm").setTypes("dongzuo");
        SearchResponse response = srb.setQuery(QueryBuilders.matchAllQuery())
                .setFetchSource(new String[]{"title", "price"}, null)
                .execute().actionGet();
        SearchHits hits = response.getHits();
        for(SearchHit hit : hits){
            System.out.println(hit.getSourceAsString());
        }
    }

    /**
     * 简单条件查询
     * eg: 查询title中 包含 "战" 的
     */
    @Test
    public void condition(){
        SearchRequestBuilder srb = client.prepareSearch("firm").setTypes("dongzuo");
        SearchResponse response = srb.setQuery(QueryBuilders.matchQuery("title", "战"))
                .execute().actionGet();
        SearchHits hits = response.getHits();
        for(SearchHit hit : hits){
            System.out.println(hit.getSourceAsString());
        }
    }

    /**
     * 条件查询 + 高亮
     */
    @Test
    public void highLight(){
        SearchRequestBuilder srb = client.prepareSearch("firm").setTypes("dongzuo");

        HighlightBuilder highlightBuilder = new HighlightBuilder();
        highlightBuilder.preTags("<h1><strong>");
        highlightBuilder.postTags("</strong><h1>");
        highlightBuilder.field("title");

        SearchResponse response = srb.setQuery(QueryBuilders.matchAllQuery())
                .setQuery(QueryBuilders.matchQuery("title", "战"))
                .setFetchSource(new String[]{"title", "price"}, null)
                .highlighter(highlightBuilder)
                .execute().actionGet();
        SearchHits hits = response.getHits();
        for(SearchHit hit : hits){
            System.out.println(hit.getSourceAsString());
            System.out.println(hit.getHighlightFields());
        }
    }


}

 

3.多条件查询,实战中主要用这个

import org.elasticsearch.action.search.SearchRequestBuilder;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.index.query.*;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
import org.junit.Test;

/**
 * 多条件查询,实战中,主要是使用这个
 * @author XXL
 * @create 2019-08-11 14:11
 */
public class MutilSearchDemo extends ClientDemo{

    /**
     * 模糊查找标题含有 “战”的,内容中必须没有“星球”的
     *
     */
    @Test
    public void searchMutil1(){
        SearchRequestBuilder srb = client.prepareSearch("firm").setTypes("dongzuo");
        QueryBuilder qb1 = QueryBuilders.matchPhraseQuery("title", "战");
        QueryBuilder qb2 = QueryBuilders.matchQuery("content","星球");
        SearchResponse response = srb.setQuery(
                QueryBuilders.boolQuery().must(qb1).mustNot(qb2)
        ).execute().actionGet();
        SearchHits hits = response.getHits();
        for(SearchHit hit : hits){
            System.out.println(hit.getSourceAsString());
        }

    }


    /**
     * should 的使用 , 提高得分
     * 标题含有战的
     * 得分:1.含有星球的得分 2. 出版日期在指定区间范围的得分
     */
    @Test
    public void shouldSearch(){
        SearchRequestBuilder srb = client.prepareSearch("firm").setTypes("dongzuo");

        MatchPhraseQueryBuilder mqb = QueryBuilders.matchPhraseQuery("title", "战");
        MatchPhraseQueryBuilder mqb2 = QueryBuilders.matchPhraseQuery("content", "星球");
        RangeQueryBuilder range = QueryBuilders.rangeQuery("publishDate").gt("2018-01-01");

        SearchResponse response = srb.setQuery(
                QueryBuilders.boolQuery().must(mqb).should(mqb2).should(range)
        ).execute().actionGet();

        SearchHits hits = response.getHits();
        for(SearchHit hit:hits){
            System.out.println(hit.getScore() + ":" +hit.getSourceAsString());
        }
    }

    /**
     * 标题含有战的
     * 过滤: 票价必须小于40的
     *
     */
    @Test
    public void filterSearch(){
        SearchRequestBuilder srb = client.prepareSearch("firm").setTypes("dongzuo");

        MatchPhraseQueryBuilder q1 = QueryBuilders.matchPhraseQuery("title", "战");
        RangeQueryBuilder priceRange = QueryBuilders.rangeQuery("price").lt(40);

        SearchResponse response = srb.setQuery(
                QueryBuilders.boolQuery().must(q1).filter(priceRange)
        ).execute().actionGet();
        SearchHits hits = response.getHits();
        for(SearchHit hit: hits){
            System.out.println(hit.getSourceAsString());
        }
    }
}

 

posted @ 2019-08-11 14:48  new_boys  阅读(3320)  评论(0编辑  收藏  举报