admin
2021-09-24 f788607ff771a47bc60d6a86e00b3433c40f3d2c
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package com.yeshi.buwan.dao.live;
 
import com.yeshi.buwan.domain.live.TVLiveChannelResourceMap;
import com.yeshi.buwan.dao.base.MongodbBaseDao;
import com.yeshi.buwan.domain.live.TVLiveResource;
import com.yeshi.buwan.query.BaseQuery;
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.data.mongodb.core.query.Update;
import org.springframework.stereotype.Repository;
 
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
 
@Repository
public class TVLiveChannelResourceMapDao extends MongodbBaseDao<TVLiveChannelResourceMap> {
 
    public void updateSelective(TVLiveChannelResourceMap bean) {
        Query query = new Query();
        Update update = new Update();
        query.addCriteria(Criteria.where("id").is(bean.getId()));
        if (bean.getChannelId() != null) {
            update.set("channelId", bean.getChannelId());
        }
        if (bean.getResource() != null) {
            update.set("resource", bean.getResource());
        }
        if (bean.getRid() != null) {
            update.set("rid", bean.getRid());
        }
        if (bean.getWeight() != null) {
            update.set("weight", bean.getWeight());
        }
        if (bean.getCreateTime() != null) {
            update.set("createTime", bean.getCreateTime());
        }
 
        if (bean.getPlayUrl() != null) {
            update.set("playUrl", bean.getPlayUrl());
        }
 
        if (bean.getSubResourceName() != null) {
            update.set("subResourceName", bean.getSubResourceName());
        }
 
        if (bean.getCreateTime() != null) {
            update.set("createTime", bean.getCreateTime());
        }
 
        update.set("updateTime", new Date());
        update(query, update);
    }
 
    private Query getQuery(DaoQuery daoQuery) {
        Query query = BaseQuery.createBaseMongoQuery(daoQuery, Arrays.asList(new Sort.Order[]{new Sort.Order(Sort.Direction.DESC, "weight"), new Sort.Order(Sort.Direction.ASC, "createTime")}));
 
        List<Criteria> andList = new ArrayList<>();
 
        if (daoQuery.channelId != null) {
            andList.add(Criteria.where("channelId").is(daoQuery.channelId));
        }
 
        if (daoQuery.resource != null) {
            andList.add(Criteria.where("resource").is(daoQuery.resource.name()));
        }
 
        if (daoQuery.rid != null) {
            andList.add(Criteria.where("rid").is(daoQuery.rid));
        }
 
        Criteria[] ands = new Criteria[andList.size()];
        andList.toArray(ands);
 
        query.addCriteria(new Criteria().andOperator(ands));
 
        return query;
    }
 
    public List<TVLiveChannelResourceMap> list(DaoQuery daoQuery) {
        Query query = getQuery(daoQuery);
        return findList(query);
    }
 
    public long count(DaoQuery daoQuery) {
        return count(getQuery(daoQuery));
    }
 
 
    public static class DaoQuery extends BaseQuery {
        public String channelId;
        public TVLiveResource resource;
        //第三方主键
        public String rid;
    }
}