diff --git a/yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/group/ImGroupServiceImpl.java b/yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/group/ImGroupServiceImpl.java index 7c05ff7..bf26a85 100644 --- a/yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/group/ImGroupServiceImpl.java +++ b/yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/group/ImGroupServiceImpl.java @@ -4,12 +4,20 @@ import org.springframework.stereotype.Service; import jakarta.annotation.Resource; import org.springframework.validation.annotation.Validated; + +import cn.iocoder.yudao.framework.common.util.collection.CollectionUtils; import cn.iocoder.yudao.module.im.controller.admin.group.vo.*; import cn.iocoder.yudao.module.im.dal.dataobject.group.ImGroupDO; +import cn.iocoder.yudao.module.im.dal.dataobject.group.ImGroupMemberDO; import cn.iocoder.yudao.framework.common.pojo.PageResult; import cn.iocoder.yudao.framework.common.util.object.BeanUtils; import cn.iocoder.yudao.module.im.dal.mysql.group.ImGroupMapper; +import cn.iocoder.yudao.module.im.dal.mysql.group.ImGroupMemberMapper; + +import java.util.Collections; +import java.util.List; +import java.util.Set; import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; import static cn.iocoder.yudao.module.im.enums.ErrorCodeConstants.*; @@ -26,6 +34,9 @@ public class ImGroupServiceImpl implements ImGroupService { @Resource private ImGroupMapper imGroupMapper; + @Resource + private ImGroupMemberMapper imGroupMemberMapper; + @Override public Long createGroup(ImGroupSaveReqVO createReqVO) { // 插入 @@ -77,4 +88,21 @@ public class ImGroupServiceImpl implements ImGroupService { return imGroupMapper.selectPage(pageReqVO); } + @Override + public List getMyGroupList(Long userId) { + // 1. 查用户所在的、仍有效的群成员记录（仅 ENABLE 状态） + List members = imGroupMemberMapper.selectEnabledListByUserId(userId); + if (members.isEmpty()) { + return Collections.emptyList(); + } + Set groupIds = CollectionUtils.convertSet(members, ImGroupMemberDO::getGroupId); + if (groupIds.isEmpty()) { + return Collections.emptyList(); + } + // 2. 查群并过滤掉已封禁/解散的 + List groups = imGroupMapper.selectByIds(groupIds); + groups.removeIf(g -> Boolean.TRUE.equals(g.getBanned()) || Boolean.TRUE.equals(g.getDissolved())); + return groups; + } + } \ No newline at end of file diff --git a/yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/message/ImGroupMessageServiceImpl.java b/yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/message/ImGroupMessageServiceImpl.java index e6df7e5..9882aad 100644 --- a/yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/message/ImGroupMessageServiceImpl.java +++ b/yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/message/ImGroupMessageServiceImpl.java @@ -49,6 +49,13 @@ import static cn.iocoder.yudao.module.im.enums.ImCommonConstants.*; @Slf4j public class ImGroupMessageServiceImpl implements ImGroupMessageService { + /** + * 仅用于规避群消息 pull 的"假空页"： + * 首批 raw 消息可能因入群时间或定向接收过滤后变成空列表，但后续更大的 id 仍然存在可见消息。 + * 因此仅在过滤结果为空时，按本轮 raw 最大 id 向后再试几次。 + */ + private static final int PULL_GROUP_MESSAGE_EMPTY_RETRY_TIMES = 3; + @Resource private ImGroupMessageMapper groupMessageMapper; @Resource @@ -100,12 +107,50 @@ public class ImGroupMessageServiceImpl implements ImGroupMessageService { if (size > MESSAGE_MAX_PULL_SIZE) { throw exception(MESSAGE_PULL_SIZE_EXCEEDED); } + // 0. 拉取时间窗：超过窗口的老消息不再通过离线通道推送 LocalDateTime minSendTime = LocalDateTime.now().minusDays(MESSAGE_GROUP_PULL_MAX_DAYS); + // 1. 拉取一批原始消息（包含当前在群、退群前补齐两部分），并过滤得到当前用户可见的结果 + List result = Collections.emptyList(); + Long currentMinId = minId; // 当前轮次的下限消息编号；初始为用户传入的 minId，后续会根据拉取结果更新 + for (int retryCount = 0; retryCount <= PULL_GROUP_MESSAGE_EMPTY_RETRY_TIMES; retryCount++) { + // 1.1 查询一批原始消息（包含当前在群、退群前补齐两部分） + Map memberMap = new HashMap<>(); // 群编号 → 群成员记录（包含当前在群、退群前补齐两部分） + List messages = getGroupMessageList(userId, currentMinId, size, minSendTime, memberMap); + if (CollUtil.isEmpty(messages)) { + break; + } + // 1.2 过滤得到当前用户可见的结果（按 id 升序） + result = filterGroupMessageList(messages, memberMap, userId, size); + // 已拿到可见消息，或原始消息已不足一页（说明没有继续重试的必要） + if (CollUtil.isNotEmpty(result) || messages.size() < size) { + break; + } + + // 1.3 更新 currentMinId，进行下一轮查询；如果本轮查询的最大消息编号没有前进，则说明后续没有更多消息了，直接 break + Long maxMessageId = getMaxValue(messages, ImGroupMessageDO::getId); + if (maxMessageId == null || maxMessageId <= currentMinId) { + break; + } + currentMinId = maxMessageId; + } + + // 2. 按当前用户补齐：消息已读态（相对 Redis 已读游标）、本人发送的回执消息的已读人数 + appendMessageStatusAndReceipt(userId, result); + log.info("[pullGroupMessageList][userId({}) minId({}) size({}) result({})]", userId, minId, size, result.size()); + return result; + } + + /** + * 查询一批原始群消息（包含当前在群、退群前补齐两部分） + */ + private List getGroupMessageList(Long userId, Long minId, Integer size, + LocalDateTime minSendTime, + Map memberMap) { // 1. 主查询：仅用"当前仍在群"的成员记录驱动（已退群的群由步骤 2 补齐） List activeMembers = groupMemberService.getActiveGroupMemberListByUserId(userId); - Map memberMap = convertMap(activeMembers, ImGroupMemberDO::getGroupId); + memberMap.putAll(convertMap(activeMembers, ImGroupMemberDO::getGroupId)); List messages = new ArrayList<>(); if (CollUtil.isNotEmpty(activeMembers)) { List groupIds = convertList(activeMembers, ImGroupMemberDO::getGroupId); @@ -128,25 +173,28 @@ public class ImGroupMessageServiceImpl implements ImGroupMessageService { for (ImGroupMemberDO quitMember : quitMembers) { List quitGroupMessages = groupMessageMapper.selectListByGroupIdAndMinIdAndQuitTimeBefore( quitMember.getGroupId(), minId, minSendTime, quitMember.getQuitTime(), size); - if (CollUtil.isNotEmpty(quitGroupMessages)) { - messages.addAll(quitGroupMessages); - memberMap.put(quitMember.getGroupId(), quitMember); + if (CollUtil.isEmpty(quitGroupMessages)) { + continue; } + messages.addAll(quitGroupMessages); + memberMap.put(quitMember.getGroupId(), quitMember); } + return messages; + } - // 3.1 按可见性过滤（入群前不可见、定向消息排除），按 id 升序后仅取本页 size 条， - // 避免「在群 + 退群前」多路合并时一次响应跨度过大、游标直接跳到全局最大 id 而漏拉中间消息 - List result = messages.stream() + /** + * 过滤一批原始群消息，得到当前用户可见的返回结果 + */ + private List filterGroupMessageList(List messages, + Map memberMap, + Long userId, Integer size) { + // 按可见性过滤（入群前不可见、定向消息排除），按 id 升序后仅取本页 size 条， + // 避免「在群 + 退群前」多路合并时一次响应跨度过大、游标直接跳到全局最大 id 而漏拉中间消息 + return messages.stream() .filter(msg -> isMessageVisible(msg, memberMap.get(msg.getGroupId()), userId)) .sorted(Comparator.comparing(ImGroupMessageDO::getId)) .limit(size) .toList(); - // 3.2 按当前用户补齐：消息已读态（相对 Redis 已读游标）、本人发送的回执消息的已读人数 - appendMessageStatusAndReceipt(userId, result); - - log.info("[pullGroupMessageList][userId({}) minId({}) size({}) result({})]", - userId, minId, size, result.size()); - return result; } /**