提交仓库

src/main/java/com/ai/bss/es/esmanage/controller/EsController.java

import com.ai.bss.es.esmanage.service.EsService;
import com.ai.bss.es.esmanage.utils.RequestEntity;
import com.ai.bss.es.esmanage.utils.ResponseResult;
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.client.RequestOptions;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.common.xcontent.XContentType;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.MatchQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.aggregations.AggregationBuilders;
import org.elasticsearch.search.aggregations.bucket.terms.TermsAggregationBuilder;
import org.elasticsearch.search.aggregations.metrics.AvgAggregationBuilder;
import org.elasticsearch.search.builder.SearchSourceBuilder;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
@RestController
@RequestMapping("/es")
public class EsController {
    @Autowired
    private RestHighLevelClient client;
    @Autowired
    private EsService esService;
    @PostMapping("/execute")
    ResponseResult execute(@RequestBody RequestEntity requestEntity) throws IOException {
//        SearchRequest requestEs = null;
//        String entity = requestEntity.getEntity();
//        String methods = requestEntity.getMethods();
//        String tableName = requestEntity.getTableName();
//        String type = requestEntity.getType();
//        if("search".equals(methods)){
//            requestEs =  new SearchRequest(tableName);
//            SearchResponse searchResponse = client.search(requestEs, RequestOptions.DEFAULT);
//
//            //6、开始处理返回的数据
//            SearchHit[] hits = searchResponse.getHits().getHits();
//            List<String> list = new ArrayList<String>();
//            for (SearchHit hit : hits) {
//                String hitString = hit.getSourceAsString();
//                System.out.println(hitString);
//                list.add(hitString);
//            }
//            ResponseResult.sucess(list);
//        }
//
//        if("GET".equals(methods)){
//            requestEs =  new SearchRequest(tableName);
//            SearchResponse searchResponse = null;//purePersist(requestEs, RequestOptions.DEFAULT);
//
//            //6、开始处理返回的数据
//            SearchHit[] hits = searchResponse.getHits().getHits();
//            List<String> list = new ArrayList<String>();
//            for (SearchHit hit : hits) {
//                String hitString = hit.getSourceAsString();
//                System.out.println(hitString);
//                list.add(hitString);
//            }
//            ResponseResult.sucess(list);
//        }
        return esService.executeEsEntity(requestEntity);
    }
    @RequestMapping("/test")
    List<String> contextLoads() throws IOException {
        //1、创建查询请求，规定查询的索引
        SearchRequest request = new SearchRequest("es_terminal_event");
        //2、创建条件构造
        SearchSourceBuilder builder = new SearchSourceBuilder();
        //3、构造条件
        MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery("address", "Lane");
        builder.query(matchQueryBuilder);
        //聚合年龄分布
        TermsAggregationBuilder ageAgg = AggregationBuilders.terms("ageAgg").field("age");
        builder.aggregation(ageAgg);
//
//        //聚合平均年龄
        AvgAggregationBuilder balanceAvg = AggregationBuilders.avg("balanceAvg").field("balance");
        builder.aggregation(balanceAvg);
        List<String> ids = new ArrayList();
        ids.add("981489");
        ids.add("155682");
        BoolQueryBuilder exersiceBoolQuery = QueryBuilders.boolQuery();
        for (String exerciseId: ids)
        {
            exersiceBoolQuery.must(QueryBuilders.termQuery("resourceId", exerciseId));
        }
        builder.query(exersiceBoolQuery);
        //4、将构造好的条件放入请求中
        request.source(builder);
        //5、开始执行发送request请求
        SearchResponse searchResponse = client.search(request, RequestOptions.DEFAULT);
        //6、开始处理返回的数据
        SearchHit[] hits = searchResponse.getHits().getHits();
        List<String> list = new ArrayList<String>();
        for (SearchHit hit : hits) {
            String hitString = hit.getSourceAsString();
            System.out.println(hitString);
            list.add(hitString);
        }
        return list;
    }
}