mongo根据字段值进行排序，类似mysql的field函数功能 mongo根据字段值进行排序，类似

背景：现有一个字段state，字段值为1，2，3，4；现在列表页需要根据字段值进行排序，要求值为2的排在前面，其他状态的数据根据修改时间倒序展示，如果是mysql数据库可以直接使用field函数进行排序，但是现在使用的数据库为mongo，没有类似函数的支持，所以采用类似case when的方式实现该功能，主要代码如下：

import org.springframework.data.domain.Page; import org.springframework.data.domain.PageImpl; import org.springframework.data.domain.PageRequest; import org.springframework.data.domain.Pageable; import org.springframework.data.domain.Sort; import org.springframework.data.mongodb.core.MongoTemplate; import org.springframework.data.mongodb.core.aggregation.Aggregation; import org.springframework.data.mongodb.core.aggregation.ConditionalOperators; import org.springframework.data.mongodb.core.aggregation.Fields; import org.springframework.data.mongodb.core.query.Criteria; import org.springframework.data.mongodb.core.query.Query; import javax.annotation.Resource; @Repository public class UserDaoImpl implements UserDao {@Resource private MongoTemplate mongoTemplate; public Page getPage(Object param) { // 根据参数拼接查询条件，getQueryCriteria为内部方法，具体实现忽略 Criteria criteria = getQueryCriteria(param); Query query = Query.query(criteria); // 使用MongoTemplate查询总数 long total = mongoTemplate.count(query, UserPO.class); int pageNo = 1; int pageSize = 10; Pageable pageable = PageRequest.of(pageNo - 1, pageSize); // 生成冗余结果字段，如果字段值为2，赋值为1，其他值赋值为0 ConditionalOperators.Cond condOperation = ConditionalOperators.when(Criteria.where("state").is(2)) .thenValueOf("1") .otherwise("0"); // 计算skip值 long skip = (pageNo - 1) * pageSize; // 获取所有字段，getBaseFields为内部方法，具体实现忽略 Fields fields = getBaseFields(); Aggregation aggregation = Aggregation.newAggregation( // 将上面的state字段命名为sortNo Aggregation.project(fields).and(condOperation).as("sortNo"), Aggregation.match(criteria), // 先根据state倒序排序，再根据修改时间倒序排序，即可实现根据state的值进行排序，类似mysql field函数排序的功能 Aggregation.sort(Sort.by(Sort.Order.desc("sortNo"), Sort.Order.desc("modifiedAt"))), Aggregation.skip(skip), Aggregation.limit(pageSize)); AggregationResults resultList = mongoTemplate.aggregate(aggregation, UserPO.class, UserPO.class); return new PageImpl<>(resultList.getMappedResults(), pageable, total); } }

【mongo根据字段值进行排序，类似mysql的field函数功能】目前还没有其他更好的方法，如果有其他方式实现，请在评论中提出。