From 5de7e6e552bfceb3d5c4dc28482833150e37b49e Mon Sep 17 00:00:00 2001
From: jiangping <jp@doumee.com>
Date: 星期三, 17 一月 2024 15:09:48 +0800
Subject: [PATCH] 调整

---
 server/service/src/main/java/com/doumee/service/business/impl/CompanyPermissionServiceImpl.java |   43 ++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 42 insertions(+), 1 deletions(-)

diff --git a/server/service/src/main/java/com/doumee/service/business/impl/CompanyPermissionServiceImpl.java b/server/service/src/main/java/com/doumee/service/business/impl/CompanyPermissionServiceImpl.java
index f2cf034..4229d86 100644
--- a/server/service/src/main/java/com/doumee/service/business/impl/CompanyPermissionServiceImpl.java
+++ b/server/service/src/main/java/com/doumee/service/business/impl/CompanyPermissionServiceImpl.java
@@ -1,10 +1,17 @@
 package com.doumee.service.business.impl;
 
+import com.doumee.core.constants.ResponseStatus;
+import com.doumee.core.exception.BusinessException;
 import com.doumee.core.model.PageData;
 import com.doumee.core.model.PageWrap;
+import com.doumee.core.utils.Constants;
 import com.doumee.core.utils.Utils;
+import com.doumee.dao.business.CompanyMapper;
 import com.doumee.dao.business.CompanyPermissionMapper;
+import com.doumee.dao.business.model.Company;
 import com.doumee.dao.business.model.CompanyPermission;
+import com.doumee.dao.system.SystemUserMapper;
+import com.doumee.dao.system.model.SystemUser;
 import com.doumee.service.business.CompanyPermissionService;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
@@ -14,6 +21,8 @@
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
+import java.util.ArrayList;
+import java.util.Date;
 import java.util.List;
 
 /**
@@ -26,10 +35,42 @@
 
     @Autowired
     private CompanyPermissionMapper companyPermissionMapper;
+    @Autowired
+    private SystemUserMapper systemUserMapper;
+    @Autowired
+    private CompanyMapper companyMapper;
 
     @Override
     public Integer create(CompanyPermission companyPermission) {
-        companyPermissionMapper.insert(companyPermission);
+        if(companyPermission.getUserId() == null || companyPermission.getCompanyIdList() == null || companyPermission.getCompanyIdList().size()==0){
+            throw  new BusinessException(ResponseStatus.BAD_REQUEST);
+        }
+        SystemUser user =systemUserMapper.selectById(companyPermission.getUserId());
+        if(user == null || (user.getDeleted() !=null && user.getDeleted())){
+            throw  new BusinessException(ResponseStatus.DATA_EMPTY.getCode(),"瀵逛笉璧凤紝鐢ㄦ埛淇℃伅涓嶅瓨鍦紝璇峰皾璇曞埛鏂伴〉闈㈤噸璇曪紒");
+        }
+        List<Company> companieList = companyMapper.selectList(new QueryWrapper<Company>().lambda()
+                .in(Company::getId,companyPermission.getCompanyIdList())
+                .eq(Company::getIsdeleted, Constants.ZERO));
+
+        if(companieList == null || companieList.size() == 0){
+            throw  new BusinessException(ResponseStatus.DATA_EMPTY.getCode(),"瀵逛笉璧凤紝璇烽�夋嫨鏈夋晥浼佷笟锛岃灏濊瘯鍒锋柊椤甸潰閲嶈瘯锛�");
+        }
+        Date date = new Date();
+        List<CompanyPermission> list = new ArrayList<>();
+        for(Company c : companieList){
+            CompanyPermission model = new CompanyPermission();
+            model.setCompanyId(c.getId());
+            model.setIsdeleted(Constants.ZERO);
+            model.setIsdeleted(Constants.ZERO);
+            model.setCreator(user.getId());
+            model.setUserId(companyPermission.getUserId());
+            model.setCreateDate(date);
+            list.add(model);
+        }
+        //鍒櫎鍘熸湁鐨勶紝鍦ㄦ壒閲忔彃鍏ユ柊鐨�
+        companyPermissionMapper.delete(new QueryWrapper<CompanyPermission>().lambda().eq(CompanyPermission::getUserId,companyPermission.getUserId()));
+        companyPermissionMapper.insertBatchSomeColumn(list);
         return companyPermission.getId();
     }
 

--
Gitblit v1.9.3