背景:現(xiàn)有一個字段state换淆,字段值為1,2几颜,3倍试,4;現(xiàn)在列表頁需要根據(jù)字段值進行排序蛋哭,要求值為2的排在前面易猫,其他狀態(tài)的數(shù)據(jù)根據(jù)修改時間倒序展示,如果是mysql數(shù)據(jù)庫可以直接使用field函數(shù)進行排序,但是現(xiàn)在使用的數(shù)據(jù)庫為mongo准颓,沒有類似函數(shù)的支持哈蝇,所以采用類似case when的方式實現(xiàn)該功能,主要代碼如下:
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<UserPO> getPage(Object param) {
// 根據(jù)參數(shù)拼接查詢條件攘已,getQueryCriteria為內(nèi)部方法炮赦,具體實現(xiàn)忽略
Criteria criteria = getQueryCriteria(param);
Query query = Query.query(criteria);
// 使用MongoTemplate查詢總數(shù)
long total = mongoTemplate.count(query, UserPO.class);
int pageNo = 1;
int pageSize = 10;
Pageable pageable = PageRequest.of(pageNo - 1, pageSize);
// 生成冗余結(jié)果字段,如果字段值為2样勃,賦值為1吠勘,其他值賦值為0
ConditionalOperators.Cond condOperation = ConditionalOperators.when(Criteria.where("state").is(2))
.thenValueOf("1")
.otherwise("0");
// 計算skip值
long skip = (pageNo - 1) * pageSize;
// 獲取所有字段,getBaseFields為內(nèi)部方法峡眶,具體實現(xiàn)忽略
Fields fields = getBaseFields();
Aggregation aggregation = Aggregation.newAggregation(
// 將上面的state字段命名為sortNo
Aggregation.project(fields).and(condOperation).as("sortNo"),
Aggregation.match(criteria),
// 先根據(jù)state倒序排序剧防,再根據(jù)修改時間倒序排序,即可實現(xiàn)根據(jù)state的值進行排序辫樱,類似mysql field函數(shù)排序的功能
Aggregation.sort(Sort.by(Sort.Order.desc("sortNo"), Sort.Order.desc("modifiedAt"))),
Aggregation.skip(skip), Aggregation.limit(pageSize));
AggregationResults<UserPO> resultList =
mongoTemplate.aggregate(aggregation, UserPO.class, UserPO.class);
return new PageImpl<>(resultList.getMappedResults(), pageable, total);
}
}
目前還沒有其他更好的方法峭拘,如果有其他方式實現(xiàn),請在評論中提出狮暑。