Elasticsearch教程（9）：RestClient操作文档Document

ES官方提供了各种不同语言的客户端，用来操作ES。这些客户端的本质就是组装DSL语句，通过http请求发送给ES。官方文档地址：https://www.elastic.co/guide/en/elasticsearch/client/index.html

其中的Java Rest Client又包括两种：

• Java Low Level Rest Client
• Java High Level Rest Client
  这里使用的是Java HighLevel Rest Client

1. 引入依赖

<dependency>
    <groupId>org.elasticsearch.client</groupId>
    <artifactId>elasticsearch-rest-high-level-client</artifactId>
</dependency>

2.初始化

    @BeforeEach
    void setUP(){
        this.client = new RestHighLevelClient(RestClient.builder(
                HttpHost.create("http://170.106.116.159:9200")
        ));
    }

    @AfterEach
    void tearTown() throws IOException {
        this.client.close();
    }

3.增删改查、批量导入

    @Test
    void testAddDocument() throws IOException {
        //根据ID查询酒店
        Hotel hotel = hotelService.getById(61083);
        //转换为文档类型
        HotelDoc hotelDoc = new HotelDoc(hotel);
        //准备request对象
        IndexRequest request = new IndexRequest("hotel").id(hotelDoc.getId().toString());
        //准备json文档
        request.source(JSON.toJSONString(hotelDoc), XContentType.JSON);
        //发送请求
        client.index(request, RequestOptions.DEFAULT);
    }

    @Test
    void testGetDocument() throws IOException {
        //准备request对象
        GetRequest request = new GetRequest("hotel").id("61083");
        //发送请求
        GetResponse response = client.get(request, RequestOptions.DEFAULT);
        //解析响应结果
        String json = response.getSourceAsString();
        HotelDoc hotelDoc = JSON.parseObject(json, HotelDoc.class);
        System.out.println(hotelDoc);
    }

    @Test
    void testUpdateDocument() throws IOException {
        //准备request对象
        UpdateRequest request = new UpdateRequest("hotel","61083");

        request.doc(
                "price", "666",
                "score", "888"
        );
         client.update(request, RequestOptions.DEFAULT);

    }

    @Test
    void testDeleteDocument() throws IOException {
        //准备request对象
        DeleteRequest request = new DeleteRequest("hotel","61083");


        client.delete(request, RequestOptions.DEFAULT);

    }

    @Test
    void testBulkDocument() throws IOException {

        List<Hotel> hotelList = hotelService.list();

        BulkRequest request = new BulkRequest();

        for (Hotel hotel : hotelList) {
            HotelDoc doc = new HotelDoc(hotel);
            request.add(new IndexRequest("hotel").id(hotel.getId().toString()).source(JSON.toJSONString(doc), XContentType.JSON));
        }

        client.bulk(request, RequestOptions.DEFAULT);

    }

4.全部代码

package cn.badianboke.hotel;

import cn.badianboke.hotel.pojo.Hotel;
import cn.badianboke.hotel.pojo.HotelDoc;
import cn.badianboke.hotel.service.IHotelService;
import com.alibaba.fastjson.JSON;
import org.apache.http.HttpHost;
import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest;
import org.elasticsearch.action.bulk.BulkRequest;
import org.elasticsearch.action.delete.DeleteRequest;
import org.elasticsearch.action.get.GetRequest;
import org.elasticsearch.action.get.GetResponse;
import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.action.update.UpdateRequest;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.client.indices.CreateIndexRequest;
import org.elasticsearch.client.indices.GetIndexRequest;
import org.elasticsearch.common.xcontent.XContentType;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import java.io.IOException;
import java.util.List;

import static cn.badianboke.hotel.constants.HotelConstants.MAPPING_TEMPLATE;

@SpringBootTest
public class HotelDocumentTest {

    @Autowired
    IHotelService hotelService;
    private RestHighLevelClient client;

    @Test
    void testAddDocument() throws IOException {
        //根据ID查询酒店
        Hotel hotel = hotelService.getById(61083);
        //转换为文档类型
        HotelDoc hotelDoc = new HotelDoc(hotel);
        //准备request对象
        IndexRequest request = new IndexRequest("hotel").id(hotelDoc.getId().toString());
        //准备json文档
        request.source(JSON.toJSONString(hotelDoc), XContentType.JSON);
        //发送请求
        client.index(request, RequestOptions.DEFAULT);
    }

    @Test
    void testGetDocument() throws IOException {
        //准备request对象
        GetRequest request = new GetRequest("hotel").id("61083");
        //发送请求
        GetResponse response = client.get(request, RequestOptions.DEFAULT);
        //解析响应结果
        String json = response.getSourceAsString();
        HotelDoc hotelDoc = JSON.parseObject(json, HotelDoc.class);
        System.out.println(hotelDoc);
    }

    @Test
    void testUpdateDocument() throws IOException {
        //准备request对象
        UpdateRequest request = new UpdateRequest("hotel","61083");

        request.doc(
                "price", "666",
                "score", "888"
        );
         client.update(request, RequestOptions.DEFAULT);

    }

    @Test
    void testDeleteDocument() throws IOException {
        //准备request对象
        DeleteRequest request = new DeleteRequest("hotel","61083");


        client.delete(request, RequestOptions.DEFAULT);

    }

    @Test
    void testBulkDocument() throws IOException {

        List<Hotel> hotelList = hotelService.list();

        BulkRequest request = new BulkRequest();

        for (Hotel hotel : hotelList) {
            HotelDoc doc = new HotelDoc(hotel);
            request.add(new IndexRequest("hotel").id(hotel.getId().toString()).source(JSON.toJSONString(doc), XContentType.JSON));
        }

        client.bulk(request, RequestOptions.DEFAULT);

    }

    @BeforeEach
    void setUP(){
        this.client = new RestHighLevelClient(RestClient.builder(
                HttpHost.create("http://170.106.116.159:9200")
        ));
    }

    @AfterEach
    void tearTown() throws IOException {
        this.client.close();
    }
}

5.Hotel

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;

@Data
@TableName("tb_hotel")
public class Hotel {
    @TableId(type = IdType.INPUT)
    private Long id;
    private String name;
    private String address;
    private Integer price;
    private Integer score;
    private String brand;
    private String city;
    private String starName;
    private String business;
    private String longitude;
    private String latitude;
    private String pic;
}

6.HotelDoc

import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@NoArgsConstructor
public class HotelDoc {
    private Long id;
    private String name;
    private String address;
    private Integer price;
    private Integer score;
    private String brand;
    private String city;
    private String starName;
    private String business;
    private String location;
    private String pic;

    public HotelDoc(Hotel hotel) {
        this.id = hotel.getId();
        this.name = hotel.getName();
        this.address = hotel.getAddress();
        this.price = hotel.getPrice();
        this.score = hotel.getScore();
        this.brand = hotel.getBrand();
        this.city = hotel.getCity();
        this.starName = hotel.getStarName();
        this.business = hotel.getBusiness();
        this.location = hotel.getLatitude() + ", " + hotel.getLongitude();
        this.pic = hotel.getPic();
    }
}

查询数据