admin
2024-01-23 81da61b828e29b7745e1382dfbbaeb685dc083ef
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package com.yeshi.fanli.dao.user;
 
import com.ks.lijin.query.BaseDaoQuery;
import com.yeshi.common.MongodbBaseDao;
import com.yeshi.fanli.entity.SystemFunction;
import com.yeshi.fanli.entity.bus.homemodule.DeviceSex;
import com.yeshi.fanli.entity.bus.user.UserFunctionsLimit;
import com.yeshi.fanli.util.db.MongoSortFactory;
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.Date;
import java.util.List;
 
@Repository
public class UserFunctionsLimitDao extends MongodbBaseDao<UserFunctionsLimit> {
 
 
    private Criteria[] getWheres(DaoQuery daoQuery) {
        List<Criteria> whereList = new ArrayList<>();
        if (daoQuery.function != null) {
            whereList.add(Criteria.where("function").is(daoQuery.function));
        }
        if (daoQuery.minCreateTime != null) {
            whereList.add(Criteria.where("createTime").gte(daoQuery.minCreateTime));
        }
 
        if (daoQuery.maxCreateTime != null) {
            whereList.add(Criteria.where("createTime").lt(daoQuery.maxCreateTime));
        }
 
        if (daoQuery.uid != null) {
            whereList.add(Criteria.where("uid").is(daoQuery.uid));
        }
 
        if (daoQuery.nowDate != null) {
            whereList.add(Criteria.where("startTime").lte(daoQuery.nowDate));
            whereList.add(Criteria.where("endTime").gt(daoQuery.nowDate));
        }
 
        Criteria[] wheres = new Criteria[whereList.size()];
        whereList.toArray(wheres);
        return wheres;
    }
 
    public List<UserFunctionsLimit> list(DaoQuery daoQuery) {
        Query query = new Query();
        query.addCriteria(new Criteria().andOperator(getWheres(daoQuery)));
        query.skip((int) daoQuery.start);
        query.limit(daoQuery.count);
 
        if (daoQuery.sortList != null && daoQuery.sortList.size() > 0)
            query.with(new Sort(MongoSortFactory.createFromSqlSort(daoQuery.sortList)));
        return findList(query);
    }
 
    public long count(DaoQuery daoQuery) {
        Query query = new Query();
        query.addCriteria(new Criteria().andOperator(getWheres(daoQuery)));
        return count(query);
    }
 
 
    public static class DaoQuery extends BaseDaoQuery {
        public Date nowDate;
        public Long uid;
        public SystemFunction function;
    }
 
}