✨ feat(group): 添加群聊成员限制及邀请好友功能

新增群聊成员人数限制，最大成员数为 500。实现邀请好友功能，确保被邀请人是当前用户的好友，若不是则返回相应错误提示。更新相关提示信息以提升用户体验。

✨ feat(group): 添加群聊成员限制及邀请好友功能
新增群聊成员人数限制，最大成员数为 500。实现邀请好友功能，确保被邀请人是当前用户的好友，若不是则返回相应错误提示。更新相关提示信息以提升用户体验。
YunaiV
1 parent 84304718
Showing 6 changed files with 78 additions and 17 deletions
yudao-module-im/yudao-module-im-api/src/main/java/cn/iocoder/yudao/module/im/enums/ErrorCodeConstants.java
yudao-module-im/yudao-module-im-api/src/main/java/cn/iocoder/yudao/module/im/enums/ImCommonConstants.java
yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/dal/mysql/friend/ImFriendMapper.java
yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/friend/ImFriendService.java
yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/friend/ImFriendServiceImpl.java
yudao-module-im/yudao-module-im-biz/src/main/java/cn/iocoder/yudao/module/im/service/group/ImGroupServiceImpl.java
@@ -37,6 +37,8 @@ public interface ErrorCodeConstants {
     ErrorCode GROUP_MEMBER_ALREADY_QUIT = new ErrorCode(1_040_500_002, "您已退出该群");
     ErrorCode GROUP_OWNER_CANNOT_QUIT = new ErrorCode(1_040_500_003, "您是群主，不可退出群聊");
     ErrorCode GROUP_CANNOT_REMOVE_SELF = new ErrorCode(1_040_500_004, "不能移除自己");
+    ErrorCode GROUP_MEMBER_EXCEED = new ErrorCode(1_040_500_005, "群聊人数不能超过{}人");
+    ErrorCode GROUP_INVITE_NOT_FRIEND = new ErrorCode(1_040_500_006, "'{}' 不是您的好友，邀请失败");
     // ========== 好友 (1-040-600-000) ==========
     ErrorCode FRIEND_NOT_EXISTS = new ErrorCode(1_040_600_000, "好友关系不存在");
@@ -13,6 +13,11 @@ public interface ImCommonConstants {
     int MESSAGE_MAX_PULL_SIZE = 1000;
     /**
+     * 群最大成员人数
+     */
+    int MAX_GROUP_MEMBER = 500;
+
+    /**
      * 消息撤回时间限制（分钟）
      */
     int MESSAGE_RECALL_TIMEOUT_MINUTES = 5;
@@ -50,14 +55,18 @@ public interface ImCommonConstants {
     /**
      * 群解散系统提示
      */
-    String GROUP_DISSOLVE_TIP_MESSAGE = "'{}'解散了群聊";
+    String GROUP_DISSOLVE_TIP_MESSAGE = "\"{}\"解散了群聊";
     /**
      * 退群系统提示
      */
-    String GROUP_QUIT_TIP_MESSAGE = "您已退出群聊";
+    String GROUP_QUIT_TIP_MESSAGE = "你已退出群聊";
     /**
      * 踢出群系统提示
      */
-    String GROUP_REMOVE_TIP_MESSAGE = "您已被移出群聊";
+    String GROUP_REMOVE_TIP_MESSAGE = "你已被移出群聊";
+    /**
+     * 邀请入群系统提示
+     */
+    String GROUP_INVITE_TIP_MESSAGE = "\"{}\"邀请\"{}\"加入了群聊";
 }
@@ -5,6 +5,7 @@ import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
 import cn.iocoder.yudao.module.im.dal.dataobject.friend.ImFriendDO;
 import org.apache.ibatis.annotations.Mapper;
+import java.util.Collection;
 import java.util.List;
 /**
@@ -26,5 +27,20 @@ public interface ImFriendMapper extends BaseMapperX&lt;ImFriendDO&gt; {
                 .eq(ImFriendDO::getUserId, userId)
                 .orderByDesc(ImFriendDO::getId));
     }
+    default List<ImFriendDO> selectListByUserIdAndStatus(Long userId, Integer status) {
+        return selectList(new LambdaQueryWrapperX<ImFriendDO>()
+                .eq(ImFriendDO::getUserId, userId)
+                .eq(ImFriendDO::getStatus, status)
+                .orderByDesc(ImFriendDO::getId));
+    }
+
+    default List<ImFriendDO> selectListByUserIdAndFriendUserIdsAndStatus(Long userId,
+                                                                        Collection<Long> friendUserIds,
+                                                                        Integer status) {
+        return selectList(new LambdaQueryWrapperX<ImFriendDO>()
+                .eq(ImFriendDO::getUserId, userId)
+                .in(ImFriendDO::getFriendUserId, friendUserIds)
+                .eq(ImFriendDO::getStatus, status));
+    }
 }
@@ -3,6 +3,7 @@ package cn.iocoder.yudao.module.im.service.friend;
 import cn.iocoder.yudao.module.im.controller.admin.friend.vo.ImFriendUpdateReqVO;
 import cn.iocoder.yudao.module.im.dal.dataobject.friend.ImFriendDO;
+import java.util.Collection;
 import java.util.List;
 /**
@@ -33,6 +34,15 @@ public interface ImFriendService {
     List<ImFriendDO> getFriendList(Long userId);
     /**
+     * 获得当前用户与指定用户之间的有效好友列表（仅 ENABLE 状态）
+     *
+     * @param userId        用户编号
+     * @param friendUserIds 需要校验的好友用户编号集合
+     * @return 有效好友关系列表
+     */
+    List<ImFriendDO> getActiveFriendList(Long userId, Collection<Long> friendUserIds);
+
+    /**
      * 查询一个好友关系记录（仅限本人视角）
      *
      * @param userId       当前登录用户编号
@@ -21,6 +21,7 @@ import org.springframework.transaction.annotation.Transactional;
 import org.springframework.validation.annotation.Validated;
 import java.time.LocalDateTime;
+import java.util.Collection;
 import java.util.List;
 import java.util.Objects;
@@ -66,6 +67,12 @@ public class ImFriendServiceImpl implements ImFriendService {
     }
     @Override
+    public List<ImFriendDO> getActiveFriendList(Long userId, Collection<Long> friendUserIds) {
+        return imFriendMapper.selectListByUserIdAndFriendUserIdsAndStatus(userId, friendUserIds,
+                CommonStatusEnum.ENABLE.getStatus());
+    }
+
+    @Override
     public ImFriendDO getFriend(Long userId, Long friendUserId) {
         return imFriendMapper.selectByUserIdAndFriendUserId(userId, friendUserId);
     }
@@ -13,14 +13,16 @@ import cn.iocoder.yudao.module.im.controller.admin.group.vo.ImGroupUpdateReqVO;
 import cn.iocoder.yudao.module.im.controller.admin.group.vo.member.ImGroupMemberInviteReqVO;
 import cn.iocoder.yudao.module.im.controller.admin.group.vo.member.ImGroupMemberRemoveReqVO;
 import cn.iocoder.yudao.module.im.controller.admin.group.vo.member.ImGroupMemberUpdateReqVO;
+import cn.iocoder.yudao.module.im.dal.dataobject.friend.ImFriendDO;
 import cn.iocoder.yudao.module.im.dal.dataobject.group.ImGroupDO;
 import cn.iocoder.yudao.module.im.dal.dataobject.group.ImGroupMemberDO;
 import cn.iocoder.yudao.module.im.dal.mysql.group.ImGroupMapper;
+import cn.iocoder.yudao.module.im.service.friend.ImFriendService;
 import cn.iocoder.yudao.module.im.service.message.ImGroupMessageService;
-import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
-import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
 import cn.iocoder.yudao.module.im.service.websocket.ImWebSocketService;
 import cn.iocoder.yudao.module.im.service.websocket.dto.ImGroupMessageDTO;
+import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
+import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
 import jakarta.annotation.Resource;
 import org.springframework.cache.annotation.CacheEvict;
 import org.springframework.cache.annotation.Cacheable;
@@ -30,11 +32,7 @@ import org.springframework.transaction.annotation.Transactional;
 import org.springframework.validation.annotation.Validated;
 import java.time.LocalDateTime;
-import java.util.Collections;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 import java.util.stream.Collectors;
 import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
@@ -66,6 +64,9 @@ public class ImGroupServiceImpl implements ImGroupService {
     private ImWebSocketService webSocketService;
     @Resource
+    private ImFriendService friendService;
+
+    @Resource
     private AdminUserApi adminUserApi;
     // ==================== 群的写操作 ====================
@@ -164,8 +165,15 @@ public class ImGroupServiceImpl implements ImGroupService {
         if (CollUtil.isEmpty(inviteReqVO.getMemberUserIds())) {
             return;
         }
-        // 1.3 校验群人数上限
+        // 1.3 校验被邀请人都是当前用户的好友
         List<Long> memberUserIds = inviteReqVO.getMemberUserIds();
+        List<ImFriendDO> friends = friendService.getActiveFriendList(userId, memberUserIds);
+        Set<Long> friendUserIds = convertSet(friends, ImFriendDO::getFriendUserId);
+        Collection<Long> notFriendUserIds = CollUtil.subtract(memberUserIds, friendUserIds);
+        if (CollUtil.isNotEmpty(notFriendUserIds)) {
+            throw exception(GROUP_INVITE_NOT_FRIEND, getUserNicknames(notFriendUserIds));
+        }
+        // 1.4 校验群人数上限
         if (activeMembers.size() + memberUserIds.size() > MAX_GROUP_MEMBER) {
             throw exception(GROUP_MEMBER_EXCEED, MAX_GROUP_MEMBER);
         }
@@ -177,12 +185,8 @@ public class ImGroupServiceImpl implements ImGroupService {
         webSocketService.sendGroupMessageAsync(new HashSet<>(memberUserIds),
                 ImGroupMessageDTO.ofGroupCreate(userId, groupId));
         // 3.2 发送邀请提示消息给所有群成员
-        AdminUserRespDTO inviter = adminUserApi.getUser(userId);
-        Map<Long, AdminUserRespDTO> invitedUserMap = adminUserApi.getUserMap(memberUserIds);
-        String invitedNames = memberUserIds.stream()
-                .map(id -> invitedUserMap.containsKey(id) ? invitedUserMap.get(id).getNickname() : String.valueOf(id))
-                .collect(Collectors.joining(","));
-        String tipContent = StrUtil.format(GROUP_INVITE_TIP_MESSAGE, inviter.getNickname(), invitedNames);
+        String tipContent = StrUtil.format(GROUP_INVITE_TIP_MESSAGE,
+                getUserNicknames(List.of(userId)), getUserNicknames(memberUserIds));
         List<ImGroupMemberDO> allMembers = groupMemberService.getActiveGroupMemberListByGroupId(groupId);
         Set<Long> allMemberUserIds = convertSet(allMembers, ImGroupMemberDO::getUserId);
         groupMessageService.sendTipGroupMessage(userId, groupId, allMemberUserIds, tipContent);
@@ -284,4 +288,17 @@ public class ImGroupServiceImpl implements ImGroupService {
         return SpringUtil.getBean(getClass());
     }
+    /**
+     * 根据用户编号集合，拼接用户昵称字符串（逗号分隔）
+     *
+     * @param userIds 用户编号集合
+     * @return 昵称字符串，如 "张三,李四"
+     */
+    private String getUserNicknames(Collection<Long> userIds) {
+        Map<Long, AdminUserRespDTO> userMap = adminUserApi.getUserMap(userIds);
+        return userIds.stream()
+                .map(id -> userMap.containsKey(id) ? userMap.get(id).getNickname() : String.valueOf(id))
+                .collect(Collectors.joining(","));
+    }
+
 }
 \ No newline at end of file