♻️ refactor(im)：用户申请列表，增加流式查询，避免一次性加载过多，或者历史无法被加载到。

YunaiV
1 parent b5f5c408
Showing 4 changed files with 47 additions and 12 deletions
yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/controller/admin/friend/ImFriendRequestController.java
yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/dal/mysql/friend/ImFriendRequestMapper.java
yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/service/friend/ImFriendRequestService.java
yudao-module-im/src/main/java/cn/iocoder/yudao/module/im/service/friend/ImFriendRequestServiceImpl.java
 package cn.iocoder.yudao.module.im.controller.admin.friend;
  
 import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.util.ObjUtil;
 import cn.iocoder.yudao.framework.common.pojo.CommonResult;
 import cn.iocoder.yudao.framework.common.util.collection.MapUtils;
 import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
@@ -15,6 +16,7 @@ import io.swagger.v3.oas.annotations.Parameter;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.annotation.Resource;
 import jakarta.validation.Valid;
+import jakarta.validation.constraints.Min;
 import jakarta.validation.constraints.NotNull;
 import jakarta.validation.constraints.Size;
 import org.springframework.validation.annotation.Validated;
@@ -75,12 +77,29 @@ public class ImFriendRequestController {
     }
  
     @GetMapping("/list")
-    @Operation(summary = "查询「我相关」的好友申请列表（含我发起的、别人加我的）")
-    public CommonResult<List<ImFriendRequestRespVO>> getMyFriendRequestList() {
-        List<ImFriendRequestDO> list = friendRequestService.getMyFriendRequestList(getLoginUserId());
+    @Operation(summary = "查询「我相关」的好友申请列表（游标分页：传 lastRequestId 加载更多）")
+    public CommonResult<List<ImFriendRequestRespVO>> getMyFriendRequestList(
+            @Parameter(description = "当前列表最旧记录的 id；首页不传")
+            @RequestParam(value = "lastRequestId", required = false) Long lastRequestId,
+            @Parameter(description = "单次拉取条数", required = true) @RequestParam("limit") @Min(1) Integer limit) {
+        List<ImFriendRequestDO> list = friendRequestService.getMyFriendRequestList(getLoginUserId(), lastRequestId, limit);
         return success(buildList(list));
     }
  
+    @GetMapping("/get")
+    @Operation(summary = "按 id 单查「我相关」的申请记录（带越权过滤；WebSocket 通知到达后用）")
+    @Parameter(name = "id", description = "申请记录编号", required = true)
+    public CommonResult<ImFriendRequestRespVO> getMyFriendRequest(@RequestParam("id") Long id) {
+        ImFriendRequestDO request = friendRequestService.getFriendRequest(id);
+        // 越权过滤：fromUser / toUser 必有一方是当前用户，否则当不存在返回 null
+        Long currentUserId = getLoginUserId();
+        if (request == null || (ObjUtil.notEqual(request.getFromUserId(), currentUserId)
+                && ObjUtil.notEqual(request.getToUserId(), currentUserId))) {
+            return success(null);
+        }
+        return success(CollUtil.getFirst(buildList(Collections.singletonList(request))));
+    }
+
     // ========== 私有方法：VO 组装 ==========
  
     private List<ImFriendRequestRespVO> buildList(List<ImFriendRequestDO> list) {
@@ -24,10 +24,14 @@ public interface ImFriendRequestMapper extends BaseMapperX&lt;ImFriendRequestDO&gt; {
                 .last("LIMIT 1"));
     }
  
-    default List<ImFriendRequestDO> selectMyList(Long userId, int limit) {
-        // 先放扩展过滤再放双向 OR；否则 .and() 返回 LambdaQueryWrapper 基类，丢失 eqIfPresent 等扩展方法
+    /**
+     * 拉取「我相关」的好友申请列表；游标分页：lastRequestId 为 null 拉首页，非 null 拉 id 严格小于它的下一页
+     */
+    default List<ImFriendRequestDO> selectMyList(Long userId, Long lastRequestId, int limit) {
+        // 先放扩展过滤再放双向 OR；否则 .and() 返回 LambdaQueryWrapper 基类，丢失 ltIfPresent 等扩展方法
         LambdaQueryWrapperX<ImFriendRequestDO> wrapper = new LambdaQueryWrapperX<>();
-        wrapper.and(w -> w.eq(ImFriendRequestDO::getFromUserId, userId)
+        wrapper.ltIfPresent(ImFriendRequestDO::getId, lastRequestId)
+                .and(w -> w.eq(ImFriendRequestDO::getFromUserId, userId)
                         .or().eq(ImFriendRequestDO::getToUserId, userId))
                 .orderByDesc(ImFriendRequestDO::getId)
                 .last("LIMIT " + limit);
@@ -39,11 +39,18 @@ public interface ImFriendRequestService {
     void refuseFriendRequest(Long userId, Long requestId, String handleContent);
  
     /**
-     * 查询「我相关」的申请列表（含我发起的、别人加我的）
+     * 查询「我相关」的申请列表（含我发起的、别人加我的）；游标分页：传 lastRequestId 拉它之前的下一页
      *
-     * @param userId 用户编号
-     * @return 申请记录列表
+     * @param userId        用户编号
+     * @param lastRequestId 当前列表最旧记录的 id；首页传 null
+     * @param limit         单次拉取条数（page size，由前端常量控制）
+     * @return 申请记录列表，按 id 倒序
      */
-    List<ImFriendRequestDO> getMyFriendRequestList(Long userId);
+    List<ImFriendRequestDO> getMyFriendRequestList(Long userId, Long lastRequestId, Integer limit);
+
+    /**
+     * 按 id 单查申请记录；通用读接口，调用方自行做越权过滤
+     */
+    ImFriendRequestDO getFriendRequest(Long id);
  
 }
@@ -183,8 +183,13 @@ public class ImFriendRequestServiceImpl implements ImFriendRequestService {
     }
  
     @Override
-    public List<ImFriendRequestDO> getMyFriendRequestList(Long userId) {
-        return friendRequestMapper.selectMyList(userId, imProperties.getFriend().getRequestListMaxSize());
+    public List<ImFriendRequestDO> getMyFriendRequestList(Long userId, Long lastRequestId, Integer limit) {
+        return friendRequestMapper.selectMyList(userId, lastRequestId, limit);
+    }
+
+    @Override
+    public ImFriendRequestDO getFriendRequest(Long id) {
+        return friendRequestMapper.selectById(id);
     }
  
     /**