From 02a98973275c0c792e5fefdacdbea272b053f886 Mon Sep 17 00:00:00 2001
From: liding <756868258@qq.com>
Date: 星期二, 28 四月 2026 11:06:00 +0800
Subject: [PATCH] feat:1.补料和补料记录 2.退料

---
 src/main/java/com/ruoyi/basic/service/impl/CustomerFollowUpFileServiceImpl.java |   22 ++++++++++++++++------
 1 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/ruoyi/basic/service/impl/CustomerFollowUpFileServiceImpl.java b/src/main/java/com/ruoyi/basic/service/impl/CustomerFollowUpFileServiceImpl.java
index 6d4d4e5..cb915c4 100644
--- a/src/main/java/com/ruoyi/basic/service/impl/CustomerFollowUpFileServiceImpl.java
+++ b/src/main/java/com/ruoyi/basic/service/impl/CustomerFollowUpFileServiceImpl.java
@@ -1,14 +1,13 @@
 package com.ruoyi.basic.service.impl;
 
 import cn.hutool.core.collection.CollUtil;
-import cn.hutool.core.stream.CollectorUtil;
 import cn.hutool.core.util.StrUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.google.common.collect.Lists;
 import com.ruoyi.basic.mapper.CustomerFollowUpFileMapper;
 import com.ruoyi.basic.pojo.CustomerFollowUpFile;
 import com.ruoyi.basic.service.CustomerFollowUpFileService;
-import com.ruoyi.basic.service.CustomerFollowUpService;
 import com.ruoyi.common.vo.SimpleFileVo;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -31,8 +30,7 @@
 @RequiredArgsConstructor
 public class CustomerFollowUpFileServiceImpl extends ServiceImpl<CustomerFollowUpFileMapper, CustomerFollowUpFile> implements CustomerFollowUpFileService {
 
-    private final CustomerFollowUpService customerFollowUpService;
-
+    private final CustomerFollowUpFileMapper customerFollowUpFileMapper;
 
     @Override
     public <T> void fillAttachment(List<T> list, Function<T, String> getAttachmentIds, BiConsumer<T, List<SimpleFileVo>> setAttachmentList) {
@@ -49,7 +47,7 @@
         List<CustomerFollowUpFile> followUpFilesByIds = new ArrayList<>();
         Lists.partition(Lists.newArrayList(ids), 999).forEach(it -> {
             followUpFilesByIds.addAll(
-                    customerFollowUpService.getFollowUpFilesByIds(it)
+                    getFollowUpFilesByIds(it)
             );
         });
         if (CollUtil.isEmpty(followUpFilesByIds)) {
@@ -64,10 +62,22 @@
             if (StrUtil.isNotBlank(attachmentIds)) {
                 List<SimpleFileVo> fileVos = Arrays.stream(attachmentIds.split(","))
                         .map(Long::valueOf)
-                        .map(it->collectMap.getOrDefault(it, (SimpleFileVo) Collections.emptyList()))
+                        .map(it->collectMap.getOrDefault(it, null))
+                        .filter(Objects::nonNull)
                         .collect(Collectors.toList());
                 setAttachmentList.accept(t, fileVos);
             }
         });
     }
+
+    private List<CustomerFollowUpFile> getFollowUpFilesByIds(Collection<Long> fileIds) {
+        if (fileIds == null || fileIds.isEmpty()) {
+            return new ArrayList<>(0);
+        }
+
+        LambdaQueryWrapper<CustomerFollowUpFile> queryWrapper = new LambdaQueryWrapper<>();
+        queryWrapper.in(CustomerFollowUpFile::getId, fileIds)
+                .select(CustomerFollowUpFile::getId, CustomerFollowUpFile::getFileUrl, CustomerFollowUpFile::getFileName);
+        return customerFollowUpFileMapper.selectList(queryWrapper);
+    }
 }

--
Gitblit v1.9.3