admin
2019-12-28 1c8b5d11b68e727ef5ccc130327e2400f937321b
fanli/src/main/java/com/yeshi/fanli/dao/elastic/ESOrderDao.java
New file
@@ -0,0 +1,65 @@
package com.yeshi.fanli.dao.elastic;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import org.elasticsearch.action.search.SearchRequest;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.common.unit.Fuzziness;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.MatchQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.index.query.TermQueryBuilder;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
import org.elasticsearch.search.builder.SearchSourceBuilder;
import org.springframework.stereotype.Repository;
import com.google.gson.Gson;
import com.yeshi.fanli.dao.ElasticBaseDao;
import com.yeshi.fanli.entity.order.ESOrder;
@Repository
public class ESOrderDao extends ElasticBaseDao<ESOrder> {
   /**
    * 查询匹配
    *
    * @param key 搜索词
    * @param uid 用户id
    * @return
    */
   public List<ESOrder> query(String key, String uid) {
      List<ESOrder> list = new ArrayList<ESOrder>();
      // 商品名称匹配
      BoolQueryBuilder should1 = QueryBuilders.boolQuery().should(QueryBuilders.matchPhraseQuery("orderNo", key))
            .should(QueryBuilders.matchPhraseQuery("goodsName", key));
      // 用户id匹配
      BoolQueryBuilder should2 = QueryBuilders.boolQuery().should(QueryBuilders.termQuery("uid", uid))
            .should(QueryBuilders.termQuery("uidDirect", uid)).should(QueryBuilders.termQuery("uidDirect", uid));
      // 查询创建
      SearchSourceBuilder sourceBuilder = new SearchSourceBuilder();
      sourceBuilder.query(QueryBuilders.boolQuery().must(should1).must(should2));
      SearchRequest searchRequest = new SearchRequest();
      searchRequest.indices(getDocument(ESOrder.class));
      searchRequest.source(sourceBuilder);
      try {
         SearchResponse searchResponse = client.search(searchRequest, RequestOptions.DEFAULT);
         SearchHits hits = searchResponse.getHits();
         SearchHit[] searchHits = hits.getHits();
         Gson gson = new Gson();
         for (SearchHit hit : searchHits) {
            String content = hit.getSourceAsString();
            list.add(gson.fromJson(content, ESOrder.class));
         }
      } catch (IOException e) {
         e.printStackTrace();
      }
      return list;
   }
}