功能完善

admin

2020-12-25 25680e135b5bdc15658622cbfde74bab73cfee77

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
package com.ks.daylucky.dao;
 
import com.ks.daylucky.pojo.DO.UserMsg;
import com.ks.daylucky.query.UserMsgQuery;
import com.ks.lib.common.dao.MongodbBaseDao;
import com.ks.lib.common.util.MongoDBQueryUtil;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Repository;
 
import java.util.ArrayList;
import java.util.List;
 
@Repository
public class UserMsgDao extends MongodbBaseDao<UserMsg> {
 
 
    public List<UserMsg> list(UserMsgQuery queryCondition) {
        List<Criteria> andList = new ArrayList<>();
        if (queryCondition.msgType != null) {
            andList.add(Criteria.where("msgType").is(queryCondition.msgType));
        }
 
        if (queryCondition.uid != null) {
            andList.add(Criteria.where("uid").is(queryCondition.uid));
        }
 
        Criteria[] ands = new Criteria[andList.size()];
        andList.toArray(ands);
        Query query = new Query();
        query.skip(queryCondition.start);
        query.limit(queryCondition.count);
        query.addCriteria(new Criteria().andOperator(ands));
        if (queryCondition.sortList != null && queryCondition.sortList.size() > 0) {
            query.with(Sort.by(MongoDBQueryUtil.convertSQLSort(queryCondition.sortList)));
        }
        return findList(query);
 
    }
 
    public long count(UserMsgQuery queryCondition) {
        List<Criteria> andList = new ArrayList<>();
        if (queryCondition.msgType != null) {
            andList.add(Criteria.where("msgType").is(queryCondition.msgType));
        }
 
        if (queryCondition.uid != null) {
            andList.add(Criteria.where("uid").is(queryCondition.uid));
        }
 
        Criteria[] ands = new Criteria[andList.size()];
        andList.toArray(ands);
        Query query = new Query();
        query.addCriteria(new Criteria().andOperator(ands));
        return count(query);
    }
 
}