From 8438d50aef0bb78e5e6c41ca1c1bc09b4e61add1 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期三, 29 七月 2026 09:18:29 +0800
Subject: [PATCH] feat(customer): 添加公海客户归属查看权限控制

---
 src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java |  102 ++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 89 insertions(+), 13 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 d9bd5da..4404387 100644
--- a/src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java
+++ b/src/main/java/com/ruoyi/basic/service/impl/CustomerServiceImpl.java
@@ -24,7 +24,9 @@
 import com.ruoyi.procurementrecord.mapper.ReturnManagementMapper;
 import com.ruoyi.procurementrecord.pojo.ReturnManagement;
 import com.ruoyi.sales.mapper.SalesLedgerMapper;
+import com.ruoyi.sales.mapper.SalesQuotationMapper;
 import com.ruoyi.sales.pojo.SalesLedger;
+import com.ruoyi.sales.pojo.SalesQuotation;
 import com.ruoyi.sales.vo.CustomerTransactionsDetailsVo;
 import com.ruoyi.sales.vo.CustomerTransactionsVo;
 import lombok.AllArgsConstructor;
@@ -53,6 +55,8 @@
 public class CustomerServiceImpl extends ServiceImpl<CustomerMapper, Customer> implements ICustomerService {
     @Autowired
     private SalesLedgerMapper salesLedgerMapper;
+    @Autowired
+    private SalesQuotationMapper salesQuotationMapper;
     @Autowired
     private ReturnManagementMapper returnManagementMapper;
     @Autowired
@@ -126,7 +130,8 @@
     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);
+        boolean isAdmin = SecurityUtils.hasPermi("customer:viewAll");
+        IPage<CustomerVo> customerPage = customerMapper.listPage(page, customer, loginUserId, isAdmin);
 
         List<CustomerVo> records = customerPage.getRecords();
         if (CollectionUtils.isEmpty(records)) {
@@ -195,7 +200,21 @@
         LoginUser loginUser = SecurityUtils.getLoginUser();
         Long tenantId = loginUser.getTenantId();
         customer.setTenantId(tenantId);
-        return customerMapper.insert(customer);
+        int result = customerMapper.insert(customer);
+
+        // 绉佹捣鏂板鏃跺悓姝ヤ竴鏉″埌鍏捣
+        if (customer.getType() != null && customer.getType() == 0) {
+            Customer publicSeaCustomer = new Customer();
+            BeanUtils.copyProperties(customer, publicSeaCustomer);
+            publicSeaCustomer.setId(null);
+            publicSeaCustomer.setType(1);
+            publicSeaCustomer.setIsAssigned(0);
+            publicSeaCustomer.setUsageUser(loginUser.getUserId());
+            publicSeaCustomer.setUsageStatus(1L);
+            customerMapper.insert(publicSeaCustomer);
+        }
+
+        return result;
     }
 
     /**
@@ -208,6 +227,7 @@
     public int updateCustomer(Customer customer) {
         LoginUser loginUser = SecurityUtils.getLoginUser();
         Long tenantId = loginUser.getTenantId();
+        validateCustomerEditPermission(customer.getId());
         customer.setTenantId(tenantId);
         return customerMapper.updateById(customer);
     }
@@ -233,6 +253,13 @@
         if (!returnManagements.isEmpty()) {
             throw new RuntimeException("瀹㈡埛妗f涓嬫湁閿�鍞��璐э紝璇峰厛鍒犻櫎閿�鍞��璐�");
         }
+
+        // 妫�鏌ユ槸鍚︽湁閿�鍞姤浠峰叧鑱�
+        List<SalesQuotation> salesQuotations = salesQuotationMapper.selectList(new QueryWrapper<SalesQuotation>().lambda().in(SalesQuotation::getCustomerId, idList));
+        if (!salesQuotations.isEmpty()) {
+            throw new RuntimeException("瀹㈡埛妗f涓嬫湁閿�鍞姤浠凤紝璇峰厛鍒犻櫎閿�鍞姤浠�");
+        }
+
         // 鏌ヨ鏄惁鏈夊凡鍒嗛厤鐨勫叕娴峰鎴�
         List<Customer> assignedPools = customerMapper.selectList(
                 new QueryWrapper<Customer>().lambda()
@@ -269,7 +296,8 @@
     public List<CustomerVo> selectCustomerLists(CustomerDto customer) {
         LoginUser loginUser = SecurityUtils.getLoginUser();
         Long loginUserId = loginUser.getUserId();
-        return customerMapper.list(customer, loginUserId);
+        boolean isAdmin = SecurityUtils.hasPermi("customer:viewAll");
+        return customerMapper.list(customer, loginUserId, isAdmin);
     }
 
     @Override
@@ -298,18 +326,27 @@
 
     @Override
     public List<Map<String, Object>> customerList(Customer customer) {
-        LambdaQueryWrapper<Customer> queryWrapper = Wrappers.lambdaQuery();
-        queryWrapper.select(Customer::getId, Customer::getCustomerName, Customer::getTaxpayerIdentificationNumber);
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        Long loginUserId = loginUser.getUserId();
+        boolean isAdmin = SecurityUtils.hasPermi("customer:viewAll");
+        CustomerDto query = new CustomerDto();
+        if (customer != null) {
+            BeanUtils.copyProperties(customer, query);
+        }
 
-        // 鑾峰彇鍘熷鏌ヨ缁撴灉
-        List<Map<String, Object>> result = customerMapper.selectMaps(queryWrapper);
+        // 澶嶇敤瀹㈡埛鍒楄〃鐨勫彲瑙佹�ц鍒欙紝閬垮厤鎶婂埆浜虹殑绉佹捣瀹㈡埛鏆撮湶缁欏綋鍓嶈处鍙�
+        List<CustomerVo> visibleCustomers = customerMapper.list(query, loginUserId, isAdmin);
+        if (CollectionUtils.isEmpty(visibleCustomers)) {
+            return Collections.emptyList();
+        }
 
-        // 灏嗕笅鍒掔嚎鍛藉悕杞崲涓洪┘宄板懡鍚�
-        return result.stream().map(map -> map.entrySet().stream()
-                .collect(Collectors.toMap(
-                        entry -> underlineToCamel(entry.getKey()),
-                        Map.Entry::getValue))
-        ).collect(Collectors.toList());
+        return visibleCustomers.stream().map(item -> {
+            Map<String, Object> map = new LinkedHashMap<>();
+            map.put("id", item.getId());
+            map.put("customerName", item.getCustomerName());
+            map.put("taxpayerIdentificationNumber", item.getTaxpayerIdentificationNumber());
+            return map;
+        }).collect(Collectors.toList());
     }
 
     // 鍒嗛厤鍏捣瀹㈡埛缁欑娴�
@@ -330,6 +367,7 @@
     public void recycleCustomer(CustomerDto customerDto) {
         Customer customer = customerMapper.selectById(customerDto.getId());
         if (customer.getType() == 1 && customer.getIsAssigned() == 1) {  // 鍏捣涓斿凡鍒嗛厤
+            validatePublicSeaRecyclePermission(customer.getId());
             customer.setIsAssigned(0);
             customer.setUsageStatus(0L);
             customer.setUsageUser(0L);
@@ -392,6 +430,44 @@
         return this.updateById(customer);
     }
 
+    private void validateCustomerEditPermission(Long customerId) {
+        Customer customer = customerMapper.selectById(customerId);
+        if (customer == null) {
+            throw new RuntimeException("瀹㈡埛妗f涓嶅瓨鍦�");
+        }
+        if (!hasSalesModuleReference(customerId)) {
+            return;
+        }
+        if (Objects.equals(customer.getType(), 0)) {
+            throw new RuntimeException("宸茶閿�鍞ā鍧楀紩鐢ㄧ殑绉佹捣瀹㈡埛涓嶈兘缂栬緫");
+        }
+        if (Objects.equals(customer.getType(), 1) && Objects.equals(customer.getIsAssigned(), 1)) {
+            throw new RuntimeException("宸查鐢ㄥ苟涓旇閿�鍞ā鍧楀紩鐢ㄧ殑鍏捣瀹㈡埛涓嶈兘缂栬緫");
+        }
+    }
+
+    private void validatePublicSeaRecyclePermission(Long customerId) {
+        if (hasSalesModuleReference(customerId)) {
+            throw new RuntimeException("宸查鐢ㄥ苟涓旇閿�鍞ā鍧楀紩鐢ㄧ殑鍏捣瀹㈡埛涓嶈兘鍥炴敹");
+        }
+    }
+
+    private boolean hasSalesModuleReference(Long customerId) {
+        Long salesLedgerCount = salesLedgerMapper.selectCount(new QueryWrapper<SalesLedger>().lambda()
+                .eq(SalesLedger::getCustomerId, customerId));
+        if (salesLedgerCount != null && salesLedgerCount > 0) {
+            return true;
+        }
+        Long salesQuotationCount = salesQuotationMapper.selectCount(new QueryWrapper<SalesQuotation>().lambda()
+                .eq(SalesQuotation::getCustomerId, customerId));
+        if (salesQuotationCount != null && salesQuotationCount > 0) {
+            return true;
+        }
+        Long returnManagementCount = returnManagementMapper.selectCount(new QueryWrapper<ReturnManagement>().lambda()
+                .eq(ReturnManagement::getCustomerId, customerId));
+        return returnManagementCount != null && returnManagementCount > 0;
+    }
+
     @Override
     public IPage<CustomerTransactionsVo> customewTransactions(Page page, String customerName) {
         return customerMapper.customewTransactions(page, customerName);

--
Gitblit v1.9.3