package org.springblade.es.test;
|
|
import org.apache.commons.lang3.StringUtils;
|
import org.apache.http.HttpHost;
|
import org.elasticsearch.action.admin.indices.refresh.RefreshRequest;
|
import org.elasticsearch.client.RestHighLevelClient;
|
import org.elasticsearch.client.RequestOptions;
|
import org.elasticsearch.client.RestClient;
|
import org.elasticsearch.client.RestClientBuilder;
|
import org.elasticsearch.action.index.IndexRequest;
|
import org.elasticsearch.action.index.IndexResponse;
|
import org.elasticsearch.action.search.SearchRequest;
|
import org.elasticsearch.action.search.SearchResponse;
|
import org.elasticsearch.index.query.QueryBuilders;
|
import org.elasticsearch.search.builder.SearchSourceBuilder;
|
import org.elasticsearch.search.SearchHit;
|
import org.elasticsearch.search.SearchHits;
|
|
import java.io.IOException;
|
import java.util.Map;
|
|
public class ElasticSearchExample {
|
|
public static void main(String[] args) throws IOException {
|
// 初始化Elasticsearch客户端
|
RestClientBuilder builder = RestClient.builder(new HttpHost("localhost", 9200, "http"));
|
RestHighLevelClient client = new RestHighLevelClient(builder);
|
|
// 索引用户数据
|
IndexRequest userIndexRequest = new IndexRequest("users_articles");
|
userIndexRequest.source("name", "John Doe", "type", "user");
|
IndexResponse userIndexResponse = client.index(userIndexRequest, RequestOptions.DEFAULT);
|
|
// 索引文章数据
|
IndexRequest articleIndexRequest = new IndexRequest("users_articles");
|
articleIndexRequest.source("title", "Elasticsearch Introduction", "type", "article");
|
IndexResponse articleIndexResponse = client.index(articleIndexRequest, RequestOptions.DEFAULT);
|
|
// 刷新索引，确保文档可搜索
|
client.indices().refresh(new RefreshRequest("users_articles"), RequestOptions.DEFAULT);
|
|
// 全文搜索
|
SearchRequest searchRequest = new SearchRequest("users_articles");
|
SearchSourceBuilder searchSourceBuilder = new SearchSourceBuilder();
|
searchSourceBuilder.query(QueryBuilders.multiMatchQuery("Elasticsearch", "name", "title"));
|
|
//设置分页
|
searchSourceBuilder.size(10);
|
searchSourceBuilder.from(0);
|
|
//将搜索资源对象设置到搜索客户端中
|
searchRequest.source(searchSourceBuilder);
|
//查询
|
SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
|
//解析搜索返回值
|
SearchHits searchHits = searchResponse.getHits();
|
//查询的总数
|
long count = searchHits.getTotalHits().value;
|
|
for (SearchHit hit : searchHits) {
|
Map<String, Object> result = hit.getSourceAsMap();
|
// 根据类型处理结果
|
String type = (String) result.get("type");
|
if ("user".equals(type)) {
|
// 处理用户名
|
String name = (String) result.get("name");
|
System.out.println("User: " + name);
|
} else if ("article".equals(type)) {
|
// 处理文章标题
|
String title = (String) result.get("title");
|
System.out.println("Article: " + title);
|
}
|
}
|
|
// 关闭客户端
|
client.close();
|
}
|
|
/**
|
* 设置高亮
|
* @param keywords
|
* @param info
|
* @return
|
*/
|
public static String setHighLight(String keywords, String info) {
|
if (StringUtils.isBlank(keywords)|| StringUtils.isBlank(info)) {
|
return info;
|
}
|
String[] keywordArray = keywords.split(" ");
|
String newInfo = info;
|
for (String keyword : keywordArray) {
|
newInfo = newInfo.replace(keyword, "<em style=\"color:red\">" + keyword + "</em>");
|
}
|
return newInfo;
|
}
|
}
|
