diff --git a/ruoyi-bussiness/src/main/java/com/ruoyi/cms/service/impl/CompanyServiceImpl.java b/ruoyi-bussiness/src/main/java/com/ruoyi/cms/service/impl/CompanyServiceImpl.java index 53e689f..5b34ab1 100644 --- a/ruoyi-bussiness/src/main/java/com/ruoyi/cms/service/impl/CompanyServiceImpl.java +++ b/ruoyi-bussiness/src/main/java/com/ruoyi/cms/service/impl/CompanyServiceImpl.java @@ -6,12 +6,10 @@ import java.io.InputStream; import java.util.*; import com.baomidou.mybatisplus.core.toolkit.Wrappers; -import com.github.pagehelper.PageHelper; import com.ruoyi.cms.domain.*; import com.ruoyi.cms.domain.query.LabelQuery; import com.ruoyi.cms.mapper.*; -import com.ruoyi.cms.service.ICompanyCollectionService; -import com.ruoyi.common.core.page.PageDomain; +import com.ruoyi.common.constant.Constants; import com.ruoyi.common.exception.ServiceException; import com.ruoyi.common.utils.SecurityUtils; import org.apache.poi.ss.usermodel.Cell; @@ -113,7 +111,19 @@ public class CompanyServiceImpl extends ServiceImpl imple if(count>1){ throw new ServiceException(company.getName()+",该公司已存在"); } - return companyMapper.updateById(company); + int i=companyMapper.updateById(company); + if(i>0){ + companyMapper.update(null,Wrappers.lambdaUpdate() + .eq(Company::getCompanyId, company.getCompanyId()) + .set(Company::getDelFlag, Constants.Del_FLAG_DELETE)); + if(Objects.isNull(company.getCompanyContactList())){return i;} + company.getCompanyContactList().forEach(x -> { + Company company1 = new Company(); + company1.setCompanyId(company.getCompanyId()); + companyMapper.insert(company1); + }); + } + return i; }