Mongodb Group by limit（分组查询和分页）功能

1.   @Override
2.   public Page<Demo> findHistoryPageList(Integer page, Integer size, String groupId, Long hcpId) {
3.     Criteria criteria = getCriteria(groupId);
4.     if (hcpId != null) {
5.       criteria.and(HCP_ID).is(hcpId);
6.     }
7.     Aggregation aggregation = Aggregation.newAggregation(
8.         CdccMedicalGroupInfo.class,
9.         Aggregation.match(criteria),
10.         Aggregation.group(GROUP_ID).first(GROUP_ID).as(GROUP_ID).first(HCP_ID).as(HCP_ID)
11.     );
12.     AggregationResults<Demo> results = mongoTemplate.aggregate(aggregation, Demo.class, Demo.class);
13.     List<Demo> groupIds = results.getMappedResults();
14.     int formIndex = page * size;
15.     int toIndex = Math.min((page + 1) * size, groupIds.size());
16.     if (formIndex > toIndex){
17.       formIndex = 0;
18.       toIndex = 0;
19.     }
20.     return new PageImpl<>(groupIds.subList(formIndex, toIndex), PageRequest.of(page, size), groupIds.size());
21.   }

1. import org.springframework.beans.factory.annotation.Autowired;
2. import org.springframework.data.mongodb.core.MongoTemplate;
3. import org.springframework.stereotype.Service;
4. import org.springframework.data.mongodb.core.aggregation.Aggregation;
5. import org.springframework.data.mongodb.core.aggregation.AggregationResults;
7. import static org.springframework.data.mongodb.core.aggregation.Aggregation.*;
9. @Service
10. public class TableAService {
12.     private final MongoTemplate mongoTemplate;
14.     @Autowired
15.     public TableAService(MongoTemplate mongoTemplate) {
16.         this.mongoTemplate = mongoTemplate;
17.     }
19.     public long getCountDistinctGroupId() {
20.         Aggregation aggregation = newAggregation(
21.             group("groupId"),
22.             group().count().as("count")
23.         );
25.         AggregationResults<CountResult> results = mongoTemplate.aggregate(
26.             aggregation, "tableA", CountResult.class
27.         );
29.         CountResult countResult = results.getUniqueMappedResult();
30.         if (countResult != null) {
31.             return countResult.getCount();
32.         } else {
33.             return 0L;
34.         }
35.     }
36. }
38. class CountResult {
39.     private long count;
41.     public long getCount() {
42.         return count;
43.     }
45.     public void setCount(long count) {
46.         this.count = count;
47.     }
48. }

1. List<Map> totalResult = mongoTemplate.aggregate(
2.     Aggregation.newAggregation(
3.         Aggregation.match(criteria),
4.         Aggregation.group("groupID"),
5.         Aggregation.group("groupID").count().as("total")
6.     ),
7.     "TableA",
8.     Map.class
9. ).getMappedResults();
10. long total = Long.valueOf(totalResult.get(0).get("total").toString());