From cf99945deda3cba4b0e634abb100847f5cc2529b Mon Sep 17 00:00:00 2001
From: huminmin <mac@MacBook-Pro.local>
Date: 星期四, 07 五月 2026 11:36:46 +0800
Subject: [PATCH] 优化共享客户以及列表查询

---
 src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java |   78 ++++++++++++++++++++++++++++++++++----
 1 files changed, 69 insertions(+), 9 deletions(-)

diff --git a/src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java b/src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java
index 959d906..c30f4f8 100644
--- a/src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java
+++ b/src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java
@@ -11,12 +11,11 @@
 import com.ruoyi.basic.dto.CustomerFollowUpDto;
 import com.ruoyi.basic.mapper.CustomerMapper;
 import com.ruoyi.basic.mapper.CustomerPrivatePoolMapper;
-import com.ruoyi.basic.pojo.Customer;
-import com.ruoyi.basic.pojo.CustomerFollowUp;
-import com.ruoyi.basic.pojo.CustomerFollowUpFile;
-import com.ruoyi.basic.pojo.CustomerPrivatePool;
+import com.ruoyi.basic.pojo.*;
+import com.ruoyi.basic.service.CustomerFollowUpFileService;
 import com.ruoyi.basic.service.CustomerFollowUpService;
 import com.ruoyi.basic.service.CustomerReturnVisitService;
+import com.ruoyi.basic.service.CustomerUserService;
 import com.ruoyi.basic.service.ICustomerService;
 import com.ruoyi.basic.vo.CustomerVo;
 import com.ruoyi.common.utils.SecurityUtils;
@@ -58,6 +57,8 @@
 
     private final CustomerFollowUpService customerFollowUpService;
     private final CustomerReturnVisitService customerReturnVisitService;
+
+    private CustomerUserService customerUserService;
 
     /**
      * 鏌ヨ瀹㈡埛妗f
@@ -115,16 +116,18 @@
      * @return 瀹㈡埛妗f
      */
     @Override
-    public IPage<CustomerDto> selectCustomerList(Page<CustomerDto> page, CustomerDto customer) {
-        IPage<CustomerDto> customerPage = customerMapper.listPage(page, customer);
+    public IPage<CustomerVo> selectCustomerList(Page<CustomerDto> page, CustomerDto customer) {
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        Long loginUserId = loginUser.getUserId();
+        IPage<CustomerVo> customerPage = customerMapper.listPage(page, customer, loginUserId);
 
-        List<CustomerDto> records = customerPage.getRecords();
+        List<CustomerVo> records = customerPage.getRecords();
         if (CollectionUtils.isEmpty(records)) {
             return customerPage;
         }
 
         List<Long> customerIds = records.stream()
-                .map(CustomerDto::getId)
+                .map(CustomerVo::getId)
                 .filter(Objects::nonNull)
                 .collect(Collectors.toList());
 
@@ -142,6 +145,16 @@
                     c.setFollowUpTime(Date.from(
                             followUp.getFollowUpTime().atZone(ZoneId.systemDefault()).toInstant()
                     ));
+                }
+
+                // 杞崲鍏变韩鐢ㄦ埛ID瀛楃涓蹭负List<Long>
+                String userIdsStr = c.getUserIdsStr();
+                if (StringUtils.isNotEmpty(userIdsStr)) {
+                    List<Long> userIds = Arrays.stream(userIdsStr.split(","))
+                            .map(String::trim)
+                            .map(Long::parseLong)
+                            .collect(Collectors.toList());
+                    c.setUserIds(userIds);
                 }
             });
         }
@@ -285,16 +298,63 @@
 
     // 鍥炴敹绉佹捣瀹㈡埛鍒板叕娴�
     @Override
+    @Transactional(rollbackFor = Exception.class)
     public void recycleCustomer(CustomerDto customerDto) {
         Customer customer = customerMapper.selectById(customerDto.getId());
         if (customer.getType() == 1 && customer.getIsAssigned() == 1) {  // 鍏捣涓斿凡鍒嗛厤
             customer.setIsAssigned(0);
             customer.setUsageStatus(0L);
-            customer.setUsageUser(null);
+            customer.setUsageUser(0L);
             customerMapper.updateById(customer);
+
+            // 鍒犻櫎璇ュ鎴风殑鎵�鏈夊叡浜叧绯�
+            customerUserService.remove(
+                new QueryWrapper<CustomerUser>().lambda()
+                    .eq(CustomerUser::getCustomerId, customerDto.getId())
+            );
         }
     }
 
+    // 瀹㈡埛鍏变韩
+    @Override
+    public void together(CustomerDto customerDto) {
+        // 鏌ヨ鐜版湁鐨勫叡浜褰�
+        List<CustomerUser> existingUsers = customerUserService.list(
+                new QueryWrapper<CustomerUser>().lambda().eq(CustomerUser::getCustomerId, customerDto.getId())
+        );
+
+        // 鑾峰彇宸插瓨鍦ㄧ殑鐢ㄦ埛ID鍒楄〃
+        List<Long> existingUserIds = existingUsers.stream()
+                .map(CustomerUser::getUserId)
+                .collect(Collectors.toList());
+
+        // 杩囨护鎺夊凡瀛樺湪鐨勭敤鎴凤紝鍙繚鐣欐柊鐢ㄦ埛
+        List<Long> newUserIds = customerDto.getUserIds().stream()
+                .filter(userId -> !existingUserIds.contains(userId))
+                .collect(Collectors.toList());
+
+        if (CollectionUtils.isEmpty(newUserIds)) {
+            return;
+        }
+
+        // 鑾峰彇褰撳墠绉熸埛ID
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        Long tenantId = loginUser.getTenantId();
+
+        // 鎵归噺淇濆瓨鏂扮殑鍏变韩璁板綍
+        List<CustomerUser> customerUsers = newUserIds.stream()
+                .map(userId -> {
+                    CustomerUser customerUser = new CustomerUser();
+                    customerUser.setCustomerId(customerDto.getId());
+                    customerUser.setUserId(userId);
+                    customerUser.setTenantId(tenantId);
+                    return customerUser;
+                })
+                .collect(Collectors.toList());
+
+        customerUserService.saveBatch(customerUsers);
+    }
+
     /**
      * 涓嬪垝绾垮懡鍚嶈浆椹煎嘲鍛藉悕
      */

--
Gitblit v1.9.3