飞天班第39节：ElasticSearch实战-2

// 客户端连接ES的实例节点
RestHighLevelClient client = new RestHighLevelClient(
        RestClient.builder(
                new HttpHost("localhost", 9200, "http"),
                new HttpHost("localhost", 9201, "http")));

SearchResponse searchResponse = restHighLevelClient.search(searchRequest, RequestOptions.DEFAULT);

// Create the low-level client
RestClient httpClient = RestClient.builder(
    new HttpHost("localhost", 9200)
).build();

// Create the HLRC
RestHighLevelClient hlrc = new RestHighLevelClientBuilder(httpClient)
    .setApiCompatibilityMode(true) 
    .build();

// Create the Java API Client with the same low level client
ElasticsearchTransport transport = new RestClientTransport(
    httpClient,
    new JacksonJsonpMapper()
);

ElasticsearchClient esClient = new ElasticsearchClient(transport);

// hlrc and esClient share the same httpClient

@SpringBootTest
public class ElasticSearchApiTest {
	@Autowired
	private RestHighLevelClient restHighLevelClient;

	// 创建索引
	@Test
	public void testCreateIndex() throws IOException {
		// 1. 创建一个请求 （相当于 PUT jude_index）
		CreateIndexRequest request = new CreateIndexRequest("jude_index");

		// 2. 通过java客户端来发送这个请求,会获得一个响应结果
		CreateIndexResponse response = restHighLevelClient.indices().create(request, RequestOptions.DEFAULT);
		System.out.println(response);
	}
  
  // 获得索引
	@Test
	void testExistsIndex() throws IOException {
		// 1、封装请求
		GetIndexRequest request = new GetIndexRequest("jude_index");
		// 2、发送请求
		boolean exists = restHighLevelClient.indices().exists(request, RequestOptions.DEFAULT);
		// 结果
		System.out.println(exists);
	}

  // 删除索引
	@Test
	void testDeleteIndex() throws IOException {
		DeleteIndexRequest request = new DeleteIndexRequest("jude_index");
		AcknowledgedResponse delete = restHighLevelClient.indices().delete(request, RequestOptions.DEFAULT);
		System.out.println(delete);
	}
  
}

// 创建文档
	@Test
	void testAddDocument() throws IOException {
		// 1、创建请求
		IndexRequest request = new IndexRequest("index_customer");
		// 2、创建对象
		Customer customer = new Customer();
		customer.setId("1005");
		customer.setAge(32);
		customer.setUsername("jude");
		customer.setNickname("我是你的神话");
		customer.setConsume(6999.99F);
		customer.setDesc("需要抽时间把所有内容都学个遍，太给力料");
		customer.setSex(1);
		customer.setBirthday("1988-07-19");
		customer.setFaceimg("https://www.icodingedu.com/img/customers/1005/logo.png");

		// 3、构建请求
		request.id("1005");
		request.timeout(TimeValue.timeValueSeconds(10));
		// 设置参数
		request.source(JSON.toJSONString(customer), XContentType.JSON);

		// 4、发送请求
		IndexResponse response = restHighLevelClient.index(request,RequestOptions.DEFAULT);
		System.out.println(response.toString());
		RestStatus status = response.status();
		System.out.println(RestStatus.OK==status|| RestStatus.CREATED == status);

	}

// 批量添加文档
@Test
void testBulkRequest() throws IOException {
  BulkRequest request = new BulkRequest();
  request.timeout("2m");

  // 从数据库查询
  ArrayList<Customer> userList = new ArrayList<>();
  userList.add(new Customer("1006",12,"test","好甜的棉花糖",10789.86F,"还不错，内容超过我的预期值，钱花的值",0,"1993-10-11","1"));
  userList.add(new Customer("1007",13,"test2","船长jack",10789.86F,"还不错，内容超过我的预期值，钱花的值",0,"1993-10-11","1"));
  userList.add(new Customer("1008",14,"test4","我爱篮球",10789.86F,"还不错，内容超过我的预期值，钱花的值",0,"1993-10-11","1"));

  for (int i = 0; i < userList.size() ; i++) {
    request.add(
      new IndexRequest("index_customer")
      .id(userList.get(i).getId())
      .source(JSON.toJSONString(userList.get(i)),XContentType.JSON)
    );
  }

  BulkResponse response = restHighLevelClient.bulk(request, RequestOptions.DEFAULT);
  System.out.println(!response.hasFailures());
}

@Test
void testIsExists() throws IOException {
  GetRequest request = new GetRequest("index_customer", "1005");
  // 过滤结果 _source
  // 不获取上下文 DO_NOT_FETCH_SOURCE
  request.fetchSourceContext(new FetchSourceContext(false));
  request.storedFields("_none_");

  boolean exists = restHighLevelClient.exists(request,RequestOptions.DEFAULT);
  System.out.println(exists);
}

@Test
void testGetDocument() throws IOException {
  GetRequest request = new GetRequest("index_customer","1005");
  GetResponse response = restHighLevelClient.get(request, RequestOptions.DEFAULT);
  System.out.println(response.toString());
  System.out.println(response.getSourceAsMap()); // 查询成功
}

@Test
void testUpdateDocument() throws IOException {
  // 1、创建请求
  UpdateRequest request = new UpdateRequest("index_customer","1005");
  request.timeout("10s");

  // 创建更新对象，设置到 doc 中！
  Customer customer = new Customer();
  customer.setNickname("我爱编程");
  request.doc(JSON.toJSONString(customer), XContentType.JSON);

  // 2、发送请求
  UpdateResponse response = restHighLevelClient.update(request, RequestOptions.DEFAULT);
  System.out.println(response.status()==RestStatus.OK);
}

@Test
void testDeleteDocument() throws IOException {
  DeleteRequest request = new DeleteRequest("index_customer", "1005");
  request.timeout("1s");
  DeleteResponse response = restHighLevelClient.delete(request, RequestOptions.DEFAULT);
  System.out.println(response.status()==RestStatus.OK);
}

    @Override
    public void deleteTodo(List<TodoEntity> dataList, String indexName) throws IOException {
        DeleteByQueryRequest deleteRequest = new DeleteByQueryRequest(indexName);
        // 查询条件
        String modelId = dataList.get(0).getModelId();
        List<String> targets = dataList.stream().map(TodoEntity::getTarget).collect(Collectors.toList());
        BoolQueryBuilder boolQuery = QueryBuilders.boolQuery().must(QueryBuilders.termQuery("modelId",modelId))
                .must(QueryBuilders.termsQuery("target",targets));
        deleteRequest.setQuery(boolQuery);
        BulkByScrollResponse bulkByScrollResponse = restHighLevelClient.deleteByQuery(deleteRequest, RequestOptions.DEFAULT);
        log.info("执行结果：{}",bulkByScrollResponse.getStatus());
    }

@Test
void testSearch() throws IOException {
  // 创建请求
  SearchRequest request = new SearchRequest("index_customer");
  // 1、构建查询对象
  SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
  searchSourceBuilder.timeout(new TimeValue(60, TimeUnit.SECONDS));
  // termQuery
  // matchAll
  // match
  // bool
  // ....
  // 编写规则（完全匹配）
  //  MatchAllQueryBuilder matchAllQueryBuilder = QueryBuilders.termQuery("name","qinjiang");
  // 2、查询条件设置
  MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery("desc","艾编程");


  // 3、查询条件放入SearchSourceBuilder
  searchSourceBuilder.query(matchQueryBuilder);

  // 发送请求
  request.source(searchSourceBuilder);

  SearchResponse response = restHighLevelClient.search(request, RequestOptions.DEFAULT);
  // 4、获取查询结果
  System.out.println(response.getHits().toString());
  System.out.println("========================================");
  for (SearchHit hit : response.getHits().getHits()) {
    System.out.println(hit.getSourceAsMap());
  }
}