后管日志私有WBS显示字段总量

blade-service/blade-manager/src/main/java/org/springblade/manager/controller/WbsTreePrivateController.java

@@ -1,5 +1,6 @@
 package org.springblade.manager.controller;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
@@ -19,13 +20,11 @@ import org.springblade.core.tool.support.Kv;
 import org.springblade.core.tool.utils.StringUtil;
 import org.springblade.manager.dto.WbsTreePrivateDTO2;
 import org.springblade.manager.dto.WbsTreePrivateDTO3;
-import org.springblade.manager.entity.ProjectInfo;
-import org.springblade.manager.entity.WbsTree;
-import org.springblade.manager.entity.WbsTreeContract;
-import org.springblade.manager.entity.WbsTreePrivate;
+import org.springblade.manager.entity.*;
 import org.springblade.manager.mapper.WbsTreeContractMapper;
 import org.springblade.manager.mapper.WbsTreePrivateMapper;
 import org.springblade.manager.service.IProjectInfoService;
+import org.springblade.manager.service.IWbsFormElementService;
 import org.springblade.manager.service.IWbsTreePrivateService;
 import org.springblade.manager.service.IWbsTreeService;
 import org.springblade.manager.vo.*;
@@ -52,6 +51,7 @@ public class WbsTreePrivateController extends BladeController {
     private final WbsTreePrivateMapper wbsTreePrivateMapper;
     private final IWbsTreeService wbsTreeService;
     private final IProjectInfoService projectInfoService;
+    private final IWbsFormElementService wbsFormElementService;
 
     /**
      * 保存项目日志划分
@@ -321,6 +321,8 @@ public class WbsTreePrivateController extends BladeController {
                     WbsTree wbsTree = wbsTreeService.getById(r.getId());
                     if (wbsTree != null){
                         r.setInitTableId(wbsTree.getInitTableId().toString());
+                        r.setElementTotal(Math.toIntExact(wbsFormElementService.count(
+                                new LambdaQueryWrapper<WbsFormElement>().eq(WbsFormElement::getFId, r.getInitTableId()))));
                     }
                 }
             }