聚合,只返回分数和personId

src/main/java/com/viontech/match/service/PersonService.java

@@ -29,14 +29,18 @@ import org.elasticsearch.index.reindex.DeleteByQueryRequest;
 import org.elasticsearch.rest.RestStatus;
 import org.elasticsearch.script.Script;
 import org.elasticsearch.script.ScriptType;
-import org.elasticsearch.search.SearchHit;
-import org.elasticsearch.search.SearchHits;
+import org.elasticsearch.search.aggregations.AggregationBuilders;
+import org.elasticsearch.search.aggregations.Aggregations;
+import org.elasticsearch.search.aggregations.BucketOrder;
+import org.elasticsearch.search.aggregations.bucket.terms.Terms;
+import org.elasticsearch.search.aggregations.bucket.terms.TermsAggregationBuilder;
+import org.elasticsearch.search.aggregations.metrics.Max;
+import org.elasticsearch.search.aggregations.metrics.MaxAggregationBuilder;
 import org.elasticsearch.search.builder.SearchSourceBuilder;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
 import java.io.IOException;
-import java.text.ParseException;
 import java.util.*;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
@@ -325,7 +329,7 @@ public class PersonService {
         if (channelSerialNums != null && channelSerialNums.size() > 0) {
             boolQuery.filter(QueryBuilders.termsQuery("channelSerialNum", channelSerialNums));
         }
-        if(person.getPersonUnid() != null && !person.getPersonUnid().equals("")){
+        if (person.getPersonUnid() != null && !person.getPersonUnid().equals("")) {
             boolQuery.filter(QueryBuilders.termQuery("personId", person.getPersonUnid()));
         }
 
@@ -345,10 +349,16 @@ public class PersonService {
 
 
         ScriptScoreQueryBuilder queryBuilder = QueryBuilders.scriptScoreQuery(boolQuery, script);
+        MaxAggregationBuilder maxScoreAgg = AggregationBuilders.max("max_score").script(new Script("_score"));
+        TermsAggregationBuilder personIdAgg = AggregationBuilders.terms("by_personId").field("personId");
+        personIdAgg.subAggregation(maxScoreAgg);
+        personIdAgg.order(BucketOrder.aggregation("max_score", false));
+
         SearchSourceBuilder builder = new SearchSourceBuilder()
                 .size(matchResultSize)
                 .query(queryBuilder)
-                .fetchSource(FETCH_SOURCE, null);
+//                .fetchSource(FETCH_SOURCE, null)
+                .aggregation(personIdAgg);
 
         return new SearchRequest(poolId).source(builder);
     }
@@ -356,32 +366,42 @@ public class PersonService {
     private List<Person> match0(SearchRequest searchRequest) throws Exception {
         List<Person> persons = new ArrayList<>();
         SearchResponse search = client.search(searchRequest, RequestOptions.DEFAULT);
-
-        SearchHits hits = search.getHits();
-        SearchHit[] hits1 = hits.getHits();
-        for (SearchHit item : hits1) {
-            Map<String, Object> source = item.getSourceAsMap();
-            Person p = new Person();
-            p.setPersonId((String) source.get("personId"));
-            p.setAge((Integer) source.get("age"));
-            p.setGender((String) source.get("gender"));
-            p.setChannelSerialNum((String) source.get("channelSerialNum"));
-            p.setBodyType((Integer) source.get("body_type"));
-
-            p.setCounttime(Optional.ofNullable((String) source.get("counttime"))
-                    .map(x -> {
-                        try {
-                            return Constant.DATE_FORMAT.get().parse(x);
-                        } catch (ParseException ignore) {
-                        }
-                        return null;
-                    })
-                    .orElse(null));
-
-            p.setScore(item.getScore());
-            p.setPersonPoolId(item.getIndex());
-            persons.add(p);
+        Aggregations aggregations = search.getAggregations();
+        Terms byPersonId = aggregations.get("by_personId");
+        List<? extends Terms.Bucket> buckets = byPersonId.getBuckets();
+        for (Terms.Bucket bucket : buckets) {
+            String personId = (String) bucket.getKey();
+            Max maxScore = bucket.getAggregations().get("max_score");
+            double value = maxScore.getValue();
+            Person person = new Person().setPersonId(personId).setScore((float) value);
+            persons.add(person);
         }
+
+//        SearchHits hits = search.getHits();
+//        SearchHit[] hits1 = hits.getHits();
+//        for (SearchHit item : hits1) {
+//            Map<String, Object> source = item.getSourceAsMap();
+//            Person p = new Person();
+//            p.setPersonId((String) source.get("personId"));
+//            p.setAge((Integer) source.get("age"));
+//            p.setGender((String) source.get("gender"));
+//            p.setChannelSerialNum((String) source.get("channelSerialNum"));
+//            p.setBodyType((Integer) source.get("body_type"));
+//
+//            p.setCounttime(Optional.ofNullable((String) source.get("counttime"))
+//                    .map(x -> {
+//                        try {
+//                            return Constant.DATE_FORMAT.get().parse(x);
+//                        } catch (ParseException ignore) {
+//                        }
+//                        return null;
+//                    })
+//                    .orElse(null));
+//
+//            p.setScore(item.getScore());
+//            p.setPersonPoolId(item.getIndex());
+//            persons.add(p);
+//        }
         return persons;
     }