diff --git a/pusong-modules/pusong-business/src/main/java/com/pusong/business/service/impl/PsContractInfoServiceImpl.java b/pusong-modules/pusong-business/src/main/java/com/pusong/business/service/impl/PsContractInfoServiceImpl.java index b53e807..5ed55d3 100644 --- a/pusong-modules/pusong-business/src/main/java/com/pusong/business/service/impl/PsContractInfoServiceImpl.java +++ b/pusong-modules/pusong-business/src/main/java/com/pusong/business/service/impl/PsContractInfoServiceImpl.java @@ -3,6 +3,7 @@ package com.pusong.business.service.impl; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; import com.baomidou.mybatisplus.core.toolkit.Wrappers; @@ -765,7 +766,13 @@ public class PsContractInfoServiceImpl implements IPsContractInfoService { * @return */ public PsContractInfo updateByCode(PsContractInfo info){ - baseMapper.update(info,Wrappers.lambdaUpdate().eq(PsContractInfo::getContractCode,info.getContractCode()).eq(PsContractInfo::getDelFlag,0)); + baseMapper.update(info, Wrappers.lambdaUpdate().eq(PsContractInfo::getContractCode,info.getContractCode()).eq(PsContractInfo::getDelFlag,0)); + if(info.getCompanyId() == null){ + LambdaUpdateWrapper eq = Wrappers.lambdaUpdate().eq(PsContractInfo::getContractCode, info.getContractCode()).eq(PsContractInfo::getDelFlag, 0); + eq.set(PsContractInfo::getCompanyId, null); + baseMapper.update(eq); + } + return info; } /**