基于spring boot的elasticsearch RestHighLevelClient的查询功能实现

由于工作中使用到了elasticsearch,现将工作中的代码贴出来,请大家指教
1、application.yml中配置elasticsearch节点信息:

elkconfig:
  clustername: test
  index: test122
  hosts:
    -
      ip: 192.168.1.2
      port: 9200
      schema: http
    -
      ip: 192.168.1.3
      port: 9200
      schema: http
    -
      ip: 192.168.1.4
      port: 9200
      schema: http

2、配置elasticsearch配置类

@Data
@ConfigurationProperties(prefix = "elkconfig")
@Configuration
public class ElkConfig {
    private String clustername;
    private String index;
    @NestedConfigurationProperty
    private List<Hosts> hosts;
}

3、单例模式实现RestHighLevelClient

@Component
public class ESRestClient {

    private static final org.slf4j.Logger logger = LoggerFactory.getLogger(ESRestClient.class);

    @Autowired
    private ElkConfig elkConfig;

    private static ESRestClient restClientUtil;

    private static RestHighLevelClient client = null;

    @PostConstruct
    public void init() {
        restClientUtil = this;
        restClientUtil.elkConfig = this.elkConfig;
    }

    public RestHighLevelClient getClient(){
        if(client != null){
            return client;
        }else {
            synchronized(ESRestClient.class) {
                /* client = new RestHighLevelClient(clientBuilder);*/
                HttpHost[] httpHosts = new HttpHost[restClientUtil.elkConfig.getHosts().size()];
                for (int i = 0; i<httpHosts.length;i++){
                    Hosts host = restClientUtil.elkConfig.getHosts().get(i);
                    httpHosts[i] = new HttpHost(host.getIp(),Integer.parseInt(host.getPort()),host.getSchema());
                }
                RestClientBuilder clientBuilder = RestClient.builder(httpHosts);
                client = new RestHighLevelClient(clientBuilder);
                logger.info("RestClientUtil intfo create rest high level client successful!");
                return client;
            }
        }
    }


}

4、新建elasticservice供调用

@Component
public class ElasticSearchService {

    private static final Logger logger = LoggerFactory.getLogger(ElasticSearchService.class);

    @Autowired
    private ESRestClient esRestClient;

    @Autowired
    private ElkConfig elkConfig;

    /**
     * 根据布尔条件进行查询
     * @param boolQueryBuilder
     * @return
     */
    public SearchResponse searchMessage(BoolQueryBuilder boolQueryBuilder) {
        try {
            RestHighLevelClient restClient = esRestClient.getClient();
            SearchRequest searchRequest = new SearchRequest(elkConfig.getIndex());

            SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();
            sourceBuilder.size(100);
            sourceBuilder.query(boolQueryBuilder);
            logger.info(sourceBuilder.toString());
            searchRequest.source(sourceBuilder);
            SearchResponse searchResponse = restClient.search(searchRequest);
            searchResponse.getHits().forEach(message -> {
                try {
                    String sourceAsString = message.getSourceAsString();
                    logger.info(sourceAsString);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });
            return searchResponse;
        } catch (IOException e) {
            e.printStackTrace();
            throw new RuntimeException("Couldn't get Detail");
        }
    }

    /**
     * 单条件检索
     * @param fieldKey
     * @param fieldValue
     * @return
     */
    public MatchPhraseQueryBuilder uniqueMatchQuery(String fieldKey,String fieldValue){
        MatchPhraseQueryBuilder matchPhraseQueryBuilder = QueryBuilders.matchPhraseQuery(fieldKey,fieldValue);
        return matchPhraseQueryBuilder;
    }

    /**
     * 多条件检索并集,适用于搜索比如包含腾讯大王卡,滴滴大王卡的用户
     * @param fieldKey
     * @param queryList
     * @return
     */
    public BoolQueryBuilder orMatchUnionWithList(String fieldKey,List<String> queryList){
        BoolQueryBuilder boolQueryBuilder = QueryBuilders.boolQuery();
        for (String fieldValue : queryList){
            boolQueryBuilder.should(QueryBuilders.matchPhraseQuery(fieldKey,fieldValue));
        }
        return boolQueryBuilder;
    }

    /**
     * 范围查询,左右都是闭集
     * @param fieldKey
     * @param start
     * @param end
     * @return
     */
    public RangeQueryBuilder rangeMathQuery(String fieldKey,String start,String end){
        RangeQueryBuilder rangeQueryBuilder = QueryBuilders.rangeQuery(fieldKey);
        rangeQueryBuilder.gte(start);
        rangeQueryBuilder.lte(end);
        return rangeQueryBuilder;
    }

    /**
     * 根据中文分词进行查询
     * @param fieldKey
     * @param fieldValue
     * @return
     */
    public MatchQueryBuilder matchQueryBuilder(String fieldKey,String fieldValue){
        MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery(fieldKey,fieldValue).analyzer("ik_smart");
        return matchQueryBuilder;
    }

}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。