个人技术分享

MongoDB 去重 分组

最近在做客服聊天系统遇到的一些小问题，记录一下。

java 项目，使用的是 MongoTemplate

1.去重

List<User> memberIds =mongoTemplate.findDistinct(query, "memberId", MSG_PREFIX, User.class);

query 查询条件

memberId 去重字段

MSG_PREFIX 查询的集合

User 转换Java类型

2.分组

使用mongoDB的聚合操作 Aggregation，对 memberId 进行分组然后取 msgId 最大的一条数据

Criteria criteriaMsg = new Criteria();
        criteriaMsg.and("memberId").in(memberIds);
        AggregationResults<ModelMap> aggregate = mongoTemplate.aggregate(Aggregation.newAggregation(
                Aggregation.match(criteriaMsg),//按条件过滤
                Aggregation.sort(Sort.by(Sort.Direction.DESC,"msgId")),//排序
                Aggregation.group("memberId").first("msgId").as("msgId"),//按会员id分组
                Aggregation.limit(10)//分页
                //将符合条件的msgId放入列表
        ), VENDOR_CONTACT_PREFIX, ModelMap.class);
        List<ModelMap> list = aggregate.getMappedResults();
        List<Long> msgIds = list.stream().map(map -> (Long) map.get("msgId")).collect(Collectors.toList());