|
|
@@ -1,6 +1,7 @@
|
|
|
package org.springblade.manager.service.impl;
|
|
|
|
|
|
import com.alibaba.fastjson.JSON;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
import lombok.AllArgsConstructor;
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
import org.apache.http.HttpEntity;
|
|
|
@@ -12,8 +13,10 @@ import org.apache.http.util.EntityUtils;
|
|
|
import org.springblade.business.feign.InformationQueryClient;
|
|
|
import org.springblade.core.mp.base.BaseServiceImpl;
|
|
|
import org.springblade.manager.entity.WbsDivide;
|
|
|
+import org.springblade.manager.entity.WbsTreeContract;
|
|
|
import org.springblade.manager.mapper.WbsDivideMapper;
|
|
|
import org.springblade.manager.service.IWbsDivideService;
|
|
|
+import org.springblade.manager.service.IWbsTreeContractService;
|
|
|
import org.springblade.manager.vo.DivideClientReq;
|
|
|
import org.springblade.manager.vo.DivideClientVo;
|
|
|
import org.springblade.manager.vo.DivideFileVo;
|
|
|
@@ -36,6 +39,7 @@ import java.util.Map;
|
|
|
public class WbsDivideServiceImpl extends BaseServiceImpl<WbsDivideMapper, WbsDivide> implements IWbsDivideService {
|
|
|
|
|
|
private final InformationQueryClient informationQueryClient;
|
|
|
+ private final IWbsTreeContractService wbsTreeContractService;
|
|
|
|
|
|
@Override
|
|
|
public List<WbsDivide> getByPKeyId(Long pKeyId) {
|
|
|
@@ -160,6 +164,28 @@ public class WbsDivideServiceImpl extends BaseServiceImpl<WbsDivideMapper, WbsDi
|
|
|
}
|
|
|
|
|
|
log.info("批量绑定节点完成,pKeyId={}, 总数={}, 成功={}", pKeyId, divideClientVos.size(), successCount);
|
|
|
+
|
|
|
+ // 更新WbsTreeContract的partitionCode属性
|
|
|
+ try {
|
|
|
+ String divideNum = divideClientVos.get(0).getDivideNum();
|
|
|
+ log.info("开始更新WbsTreeContract的partitionCode,pKeyId={}, divideNum={}", pKeyId, divideNum);
|
|
|
+
|
|
|
+ WbsTreeContract wbsNode = wbsTreeContractService.getOne(
|
|
|
+ Wrappers.<WbsTreeContract>lambdaQuery().eq(WbsTreeContract::getPKeyId, pKeyId)
|
|
|
+ );
|
|
|
+
|
|
|
+ if (wbsNode != null) {
|
|
|
+ wbsNode.setPartitionCode(divideNum);
|
|
|
+ boolean updateResult = wbsTreeContractService.updateById(wbsNode);
|
|
|
+ log.info("更新WbsTreeContract的partitionCode完成,pKeyId={}, divideNum={}, 结果={}",
|
|
|
+ pKeyId, divideNum, updateResult);
|
|
|
+ } else {
|
|
|
+ log.warn("未找到pKeyId={}对应的WbsTreeContract记录", pKeyId);
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("更新WbsTreeContract的partitionCode失败,pKeyId={}, error={}", pKeyId, e.getMessage(), e);
|
|
|
+ }
|
|
|
+
|
|
|
return successCount;
|
|
|
}
|
|
|
|