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 |  234 +++++++++++++++++++++++++++++++++++++++------------------
 1 files changed, 159 insertions(+), 75 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 b7cb426..c30f4f8 100644
--- a/src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java
+++ b/src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java
@@ -10,33 +10,34 @@
 import com.ruoyi.basic.dto.CustomerDto;
 import com.ruoyi.basic.dto.CustomerFollowUpDto;
 import com.ruoyi.basic.mapper.CustomerMapper;
-import com.ruoyi.basic.pojo.Customer;
-import com.ruoyi.basic.pojo.CustomerFollowUp;
-import com.ruoyi.basic.pojo.CustomerFollowUpFile;
+import com.ruoyi.basic.mapper.CustomerPrivatePoolMapper;
+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;
 import com.ruoyi.common.utils.StringUtils;
+import com.ruoyi.common.utils.bean.BeanUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.framework.security.LoginUser;
 import com.ruoyi.framework.web.domain.AjaxResult;
-import com.ruoyi.project.system.domain.SysUser;
 import com.ruoyi.sales.mapper.SalesLedgerMapper;
 import com.ruoyi.sales.pojo.SalesLedger;
-import lombok.AllArgsConstructor;
+import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
-import org.springframework.beans.BeanUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
 import org.springframework.web.multipart.MultipartFile;
 
-import java.util.Arrays;
-import java.util.List;
-import java.util.Map;
-import java.util.Objects;
+
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+import java.time.ZoneId;
+import java.util.*;
 import java.util.stream.Collectors;
 
 
@@ -47,17 +48,17 @@
  * @date 2025-05-07
  */
 @Service
-@AllArgsConstructor
+@RequiredArgsConstructor
 @Slf4j
 public class CustomerServiceImpl extends ServiceImpl<CustomerMapper, Customer> implements ICustomerService {
     private final SalesLedgerMapper salesLedgerMapper;
-    private CustomerMapper customerMapper;
+    private final CustomerPrivatePoolMapper customerPrivatePoolMapper;
+    private final CustomerMapper customerMapper;
 
-    private CustomerFollowUpService customerFollowUpService;
+    private final CustomerFollowUpService customerFollowUpService;
+    private final CustomerReturnVisitService customerReturnVisitService;
 
-    private CustomerFollowUpFileService customerFollowUpFileService;
-
-    private CustomerReturnVisitService customerReturnVisitService;
+    private CustomerUserService customerUserService;
 
     /**
      * 鏌ヨ瀹㈡埛妗f
@@ -77,14 +78,9 @@
      * @return 瀹㈡埛璇︽儏DTO
      */
     @Override
-    public CustomerDto selectCustomerDetailById(Long id) {
-        Customer customer = customerMapper.selectById(id);
-        if (customer == null) {
-            return null;
-        }
-
-        CustomerDto dto = new CustomerDto();
-        BeanUtils.copyProperties(customer, dto);
+    public CustomerVo selectCustomerDetailById(Long id) {
+        CustomerVo customerVo = new CustomerVo();
+        BeanUtils.copyProperties(this.getById(id), customerVo);
 
         // 鏌ヨ璺熻繘璁板綍
         List<CustomerFollowUp> followUpList = customerFollowUpService.list(
@@ -92,8 +88,7 @@
                         .eq(CustomerFollowUp::getCustomerId, id)
                         .orderByDesc(CustomerFollowUp::getFollowUpTime)
         );
-
-        if (!CollectionUtils.isEmpty(followUpList)) {
+        if (!com.baomidou.mybatisplus.core.toolkit.CollectionUtils.isEmpty(followUpList)) {
             List<CustomerFollowUpDto> followUpDtoList = followUpList.stream().map(followUp -> {
                 CustomerFollowUpDto followUpDto = new CustomerFollowUpDto();
                 BeanUtils.copyProperties(followUp, followUpDto);
@@ -108,10 +103,10 @@
                 return followUpDto;
             }).collect(Collectors.toList());
 
-            dto.setFollowUpList(followUpDtoList);
+            customerVo.setFollowUpList(followUpDtoList);
         }
 
-        return dto;
+        return customerVo;
     }
 
     /**
@@ -121,58 +116,65 @@
      * @return 瀹㈡埛妗f
      */
     @Override
-    public IPage<Customer> selectCustomerList(Page<Customer> page, Customer customer) {
-        // 1. 澶勭悊绌哄�煎満鏅紙鍙傛暟鏍¢獙锛�
-        if (page == null) {
-            page = Page.of(1, 10); // 榛樿绗�1椤碉紝姣忛〉10鏉℃暟鎹�
-        }
-        if (customer == null) {
-            customer = new Customer(); // 閬垮厤绌哄璞″鑷寸殑NPE
+    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<CustomerVo> records = customerPage.getRecords();
+        if (CollectionUtils.isEmpty(records)) {
+            return customerPage;
         }
 
-        // 2. 鏋勫缓鏌ヨ鏉′欢锛堝寮虹┖鍊煎畨鍏級
-        LambdaQueryWrapper<Customer> queryWrapper = new LambdaQueryWrapper<>();
-        String customerName = customer.getCustomerName();
-        String customerType = customer.getCustomerType();
-        if (StringUtils.isNotBlank(customerName)) {
-            queryWrapper.like(Customer::getCustomerName, customerName);
-        }
-        if (StringUtils.isNotBlank(customerType)) {
-            queryWrapper.like(Customer::getCustomerType, customerType);
-        }
-
-        // 3. 鎵ц鍒嗛〉鏌ヨ锛堜繚鐣欏垎椤靛厓鏁版嵁锛�
-        IPage<Customer> customerPage = customerMapper.selectPage(page, queryWrapper);
-
-        // 4. 鏁版嵁澶勭悊锛堝寮虹┖鍊煎畨鍏� & 浠g爜鍙鎬э級
-        List<Customer> processedList = customerPage.getRecords().stream()
-                .filter(Objects::nonNull) // 杩囨护绌哄璞★紙閬垮厤鍚庣画鎿嶄綔NPE锛�
-                .peek(c -> {
-                    // 瀹夊叏鑾峰彇瀛楁锛岄伩鍏峮ull鍊兼嫾鎺�
-                    String address = StringUtils.defaultString(c.getCompanyAddress(), "");
-                    String phone = StringUtils.defaultString(c.getCompanyPhone(), "");
-                    c.setAddressPhone(address + "(" + phone + ")");
-
-                    // 鏌ヨ鏈�鏂扮殑璺熻繘璁板綍
-                    CustomerFollowUp followUp = customerFollowUpService.getOne(
-                            new LambdaQueryWrapper<CustomerFollowUp>()
-                                    .eq(CustomerFollowUp::getCustomerId, c.getId())
-                                    .orderByDesc(CustomerFollowUp::getFollowUpTime)
-                                    .last("LIMIT 1")
-                    );
-
-                    if (followUp != null) {
-                        c.setFollowUpLevel(followUp.getFollowUpLevel());
-                        c.setFollowUpTime(followUp.getFollowUpTime());
-                    }
-                })
+        List<Long> customerIds = records.stream()
+                .map(CustomerVo::getId)
+                .filter(Objects::nonNull)
                 .collect(Collectors.toList());
 
-        // 5. 鏇存柊鍒嗛〉缁撴灉涓殑鏁版嵁锛堜繚鎸佸垎椤典俊鎭畬鏁达級
-        IPage<Customer> resultPage = new Page<>(customerPage.getCurrent(), customerPage.getSize(), customerPage.getTotal());
-        resultPage.setRecords(processedList);
+        if (!CollectionUtils.isEmpty(customerIds)) {
+            Map<Long, CustomerFollowUp> latestFollowUpMap = getLatestFollowUpMap(customerIds);
 
-        return customerPage; // 杩斿洖鍖呭惈鍒嗛〉淇℃伅鐨処Page瀵硅薄
+            records.forEach(c -> {
+                String address = StringUtils.defaultString(c.getCompanyAddress(), "");
+                String phone = StringUtils.defaultString(c.getCompanyPhone(), "");
+                c.setAddressPhone(address + "(" + phone + ")");
+
+                CustomerFollowUp followUp = latestFollowUpMap.get(c.getId());
+                if (followUp != null) {
+                    c.setFollowUpLevel(followUp.getFollowUpLevel());
+                    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);
+                }
+            });
+        }
+
+        return customerPage;
+    }
+
+    private Map<Long, CustomerFollowUp> getLatestFollowUpMap(List<Long> customerIds) {
+        List<CustomerFollowUp> followUps = customerFollowUpService.list(
+                new LambdaQueryWrapper<CustomerFollowUp>()
+                        .in(CustomerFollowUp::getCustomerPrivatePoolId, customerIds)
+                        .orderByDesc(CustomerFollowUp::getFollowUpTime)
+        );
+
+        return followUps.stream()
+                .collect(Collectors.toMap(
+                        CustomerFollowUp::getCustomerPrivatePoolId,
+                        followUp -> followUp,
+                        (existing, replacement) -> existing
+                ));
     }
 
     /**
@@ -217,11 +219,22 @@
         if (!salesLedgers.isEmpty()) {
             throw new RuntimeException("瀹㈡埛妗f涓嬫湁閿�鍞悎鍚岋紝璇峰厛鍒犻櫎閿�鍞悎鍚�");
         }
+        // 鏌ヨ鏄惁鏈夊凡鍒嗛厤鐨勫叕娴峰鎴�
+        List<Customer> assignedPools = customerMapper.selectList(
+                new QueryWrapper<Customer>().lambda()
+                        .in(Customer::getId, idList)
+                        .eq(Customer::getType, 1).
+                        eq(Customer::getIsAssigned, 1)  // 鍏捣瀹㈡埛
+        );
+        if (!assignedPools.isEmpty()) {
+            throw new RuntimeException("瀹㈡埛妗f涓嬫湁宸插垎閰嶇殑鍏捣瀹㈡埛锛岃鍏堟敹鍥�");
+        }
         //  鍒犻櫎瀹㈡埛鐨勫悓鏃朵篃闇�瑕佸垹闄ゅ搴旂殑瀹㈡埛璺熼殢銆侀檮浠跺拰鍥炶鎻愰啋
         for (Long id : ids) {
             customerFollowUpService.deleteByCustomerId(id);
             customerReturnVisitService.deleteByCustomerId(id);
         }
+        customerMapper.delete(new QueryWrapper<Customer>().lambda().in(Customer::getId, idList));
 
         return customerMapper.deleteBatchIds(idList);
     }
@@ -271,6 +284,77 @@
         ).collect(Collectors.toList());
     }
 
+    // 鍒嗛厤鍏捣瀹㈡埛缁欑娴�
+    @Override
+    public void assignCustomer(CustomerDto customerDto) {
+        Customer customer = customerMapper.selectById(customerDto.getId());
+        if (customer.getType() == 1 && customer.getIsAssigned() == 0) {  // 鍏捣涓斿彲鍒嗛厤
+            customer.setIsAssigned(1);
+            customer.setUsageStatus(1L);
+            customer.setUsageUser(customerDto.getUsageUser());
+            customerMapper.updateById(customer);
+        }
+    }
+
+    // 鍥炴敹绉佹捣瀹㈡埛鍒板叕娴�
+    @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(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