Merge remote-tracking branch 'origin/dev_business' into dev_business

liyong

8 小时以前 7a23c450f3ac85de7dca1b908de273ff636ce218

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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package cn.iocoder.yudao.module.im.dal.mysql.group;
 
import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.im.dal.dataobject.group.ImGroupMemberDO;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import org.apache.ibatis.annotations.Mapper;
 
import java.time.LocalDateTime;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 
/**
 * IM 群成员 Mapper
 *
 * @author 芋道源码
 */
@Mapper
public interface ImGroupMemberMapper extends BaseMapperX<ImGroupMemberDO> {
 
    default List<ImGroupMemberDO> selectListByGroupId(Long groupId) {
        return selectList(new LambdaQueryWrapperX<ImGroupMemberDO>().eq(ImGroupMemberDO::getGroupId, groupId));
    }
 
    default ImGroupMemberDO selectByGroupIdAndUserId(Long groupId, Long userId) {
        return selectOne(new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getGroupId, groupId)
                .eq(ImGroupMemberDO::getUserId, userId));
    }
 
    default List<ImGroupMemberDO> selectListByGroupIdAndUserIds(Long groupId, Collection<Long> userIds) {
        return selectList(new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getGroupId, groupId)
                .in(ImGroupMemberDO::getUserId, userIds));
    }
 
    default List<ImGroupMemberDO> selectListByGroupIdAndStatus(Long groupId, Integer status) {
        return selectList(new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getGroupId, groupId)
                .eq(ImGroupMemberDO::getStatus, status));
    }
 
    default List<ImGroupMemberDO> selectListByGroupIdAndStatusAndRoles(Long groupId, Integer status,
                                                                        Collection<Integer> roles) {
        return selectList(new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getGroupId, groupId)
                .eq(ImGroupMemberDO::getStatus, status)
                .in(ImGroupMemberDO::getRole, roles));
    }
 
    default List<ImGroupMemberDO> selectListByUserIdAndStatus(Long userId, Integer status) {
        return selectList(new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getUserId, userId)
                .eq(ImGroupMemberDO::getStatus, status));
    }
 
    default List<ImGroupMemberDO> selectListByUserId(Long userId) {
        return selectList(ImGroupMemberDO::getUserId, userId);
    }
 
    @SuppressWarnings("UnusedReturnValue")
    default int updateByGroupIdAndStatus(Long groupId, Integer oldStatus, ImGroupMemberDO updateObj) {
        return update(updateObj, new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getGroupId, groupId)
                .eq(ImGroupMemberDO::getStatus, oldStatus));
    }
 
    @SuppressWarnings("UnusedReturnValue")
    default int updateByGroupIdAndUserIdsAndStatus(Long groupId, Collection<Long> userIds,
                                                   Integer oldStatus, ImGroupMemberDO updateObj) {
        return update(updateObj, new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getGroupId, groupId)
                .in(ImGroupMemberDO::getUserId, userIds)
                .eq(ImGroupMemberDO::getStatus, oldStatus));
    }
 
    @SuppressWarnings("UnusedReturnValue")
    default int updateListByGroupIdAndUserIds(Long groupId, Collection<Long> userIds, ImGroupMemberDO updateObj) {
        return update(updateObj, new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getGroupId, groupId)
                .in(ImGroupMemberDO::getUserId, userIds));
    }
 
    default Long selectCountByGroupIdAndRoleAndStatus(Long groupId, Integer role, Integer status) {
        return selectCount(new LambdaQueryWrapperX<ImGroupMemberDO>()
                .eq(ImGroupMemberDO::getGroupId, groupId)
                .eq(ImGroupMemberDO::getRole, role)
                .eq(ImGroupMemberDO::getStatus, status));
    }
 
    /**
     * 批量按 group_id 统计指定状态的成员数
     */
    default Map<Long, Long> selectCountMapByGroupIdsAndStatus(Collection<Long> groupIds, Integer status) {
        if (CollUtil.isEmpty(groupIds)) {
            return Collections.emptyMap();
        }
        List<Map<String, Object>> rows = selectMaps(Wrappers.<ImGroupMemberDO>query()
                .select("group_id AS groupId", "COUNT(*) AS cnt")
                .in("group_id", groupIds)
                .eq("status", status)
                .groupBy("group_id"));
        // 转换成 Map<Long, Long>
        Map<Long, Long> result = new HashMap<>(rows.size());
        rows.forEach(row -> result.put(
                ((Number) row.get("groupId")).longValue(),
                ((Number) row.get("cnt")).longValue()));
        return result;
    }
 
    @SuppressWarnings("UnusedReturnValue")
    default int updateMuteEndTimeNull(Long id) {
        return update(null, Wrappers.<ImGroupMemberDO>lambdaUpdate()
                .eq(ImGroupMemberDO::getId, id)
                .set(ImGroupMemberDO::getMuteEndTime, null));
    }
 
    @SuppressWarnings("UnusedReturnValue")
    default int updateRejoinFields(Long id, Integer status, LocalDateTime joinTime,
                                   Integer role, Integer addSource, Long inviterUserId) {
        return update(null, Wrappers.<ImGroupMemberDO>lambdaUpdate()
                .eq(ImGroupMemberDO::getId, id)
                .set(ImGroupMemberDO::getStatus, status)
                .set(ImGroupMemberDO::getJoinTime, joinTime)
                .set(ImGroupMemberDO::getRole, role)
                .set(ImGroupMemberDO::getAddSource, addSource)
                .set(ImGroupMemberDO::getInviterUserId, inviterUserId)
                .set(ImGroupMemberDO::getQuitTime, null)
                .set(ImGroupMemberDO::getMuteEndTime, null)
                .set(ImGroupMemberDO::getUpdateTime, joinTime));
    }
 
}