diff --git a/yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/dal/mysql/message/ImChannelMessageMapper.java b/yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/dal/mysql/message/ImChannelMessageMapper.java index 6f1cd1a..a847d6c 100644 --- a/yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/dal/mysql/message/ImChannelMessageMapper.java +++ b/yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/dal/mysql/message/ImChannelMessageMapper.java @@ -8,6 +8,7 @@ import cn.iocoder.yudao.module.im.controller.admin.manager.message.vo.channel.Im import cn.iocoder.yudao.module.im.dal.dataobject.message.ImChannelMessageDO; import org.apache.ibatis.annotations.Mapper; +import java.util.Collections; import java.util.List; /** @@ -32,7 +33,7 @@ public interface ImChannelMessageMapper extends BaseMapperX return selectList(new LambdaQueryWrapperX() .gt(ImChannelMessageDO::getId, minId) .and(w -> w.isNull(ImChannelMessageDO::getReceiverUserIds) - .or().eq(ImChannelMessageDO::getReceiverUserIds, "") + .or().eq(ImChannelMessageDO::getReceiverUserIds, Collections.emptyList()) .or().apply(MyBatisUtils.findInSet("receiver_user_ids"), userId)) .orderByAsc(ImChannelMessageDO::getId) .last("LIMIT " + size)); diff --git a/yudao-module-im/src/test/java/cn/iocoder/yudao/module/im/dal/mysql/message/ImChannelMessageMapperTest.java b/yudao-module-im/src/test/java/cn/iocoder/yudao/module/im/dal/mysql/message/ImChannelMessageMapperTest.java new file mode 100644 index 0000000..ee28369 --- /dev/null +++ b/yudao-module-im/src/test/java/cn/iocoder/yudao/module/im/dal/mysql/message/ImChannelMessageMapperTest.java @@ -0,0 +1,59 @@ +package cn.iocoder.yudao.module.im.dal.mysql.message; + +import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest; +import cn.iocoder.yudao.module.im.dal.dataobject.message.ImChannelMessageDO; +import jakarta.annotation.Resource; +import org.junit.jupiter.api.Test; +import org.springframework.test.context.jdbc.Sql; + +import java.time.LocalDateTime; +import java.util.List; + +import static org.junit.jupiter.api.Assertions.assertEquals; + +@Sql(statements = "DELETE FROM im_channel_message", executionPhase = Sql.ExecutionPhase.AFTER_TEST_METHOD) +public class ImChannelMessageMapperTest extends BaseDbUnitTest { + + @Resource + private ImChannelMessageMapper mapper; + + @Test + public void testSelectListByUserAndMinId_includesBroadcastAndMatchingRecipients() { + ImChannelMessageDO nullRecipients = insertMessage(null); + ImChannelMessageDO emptyRecipients = insertMessage(List.of()); + ImChannelMessageDO matchingRecipients = insertMessage(List.of(1L, 2L)); + insertMessage(List.of(2L, 11L)); // 用户 1 不应误命中用户 11 + ImChannelMessageDO deleted = insertMessage(List.of(1L)); + mapper.deleteById(deleted.getId()); + + List result = mapper.selectListByUserAndMinId(1L, 0L, 100); + + assertEquals(List.of(nullRecipients.getId(), emptyRecipients.getId(), matchingRecipients.getId()), + result.stream().map(ImChannelMessageDO::getId).toList()); + } + + @Test + public void testSelectListByUserAndMinId_respectsCursorAndLimit() { + ImChannelMessageDO cursor = insertMessage(null); + ImChannelMessageDO next = insertMessage(List.of()); + insertMessage(List.of(1L)); + + List result = mapper.selectListByUserAndMinId(1L, cursor.getId(), 1); + + assertEquals(List.of(next.getId()), result.stream().map(ImChannelMessageDO::getId).toList()); + } + + private ImChannelMessageDO insertMessage(List receiverUserIds) { + ImChannelMessageDO message = ImChannelMessageDO.builder() + .channelId(10L) + .materialId(20L) + .type(1) + .content("{}") + .receiverUserIds(receiverUserIds) + .sendTime(LocalDateTime.now()) + .build(); + mapper.insert(message); + return message; + } + +}