liuyc 2 ani în urmă
părinte
comite
b7a3b9e822

+ 4 - 3
blade-service/blade-control/src/main/java/org/springblade/control/service/impl/TaskProcessServiceImpl.java

@@ -31,6 +31,7 @@ import org.springblade.core.tool.utils.BeanUtil;
 import org.springblade.core.tool.utils.DateUtil;
 import org.springblade.core.tool.utils.Func;
 import org.springblade.core.tool.utils.ObjectUtil;
+import org.springblade.manager.entity.WbsTreeContract;
 import org.springblade.resource.feign.IOSSClient;
 import org.springblade.system.entity.Dept;
 import org.springblade.system.feign.ISysClient;
@@ -1054,9 +1055,9 @@ public class TaskProcessServiceImpl extends BaseServiceImpl<TaskProcessMapper, T
                     //已完成 = approve = 1
                     jdbcTemplate.execute("update c_project_cost_budget set approve = 1 where id in(" + dataId + ")");
 
-                    ProjectCostBudget projectCostBudget = jdbcTemplate.query("select * from c_project_cost_budget where id = " + dataId, new BeanPropertyRowMapper<>(ProjectCostBudget.class)).stream().findAny().orElse(null);
-                    if (projectCostBudget != null && ObjectUtil.isNotEmpty(projectCostBudget.getProjectId())) {
-                        jdbcTemplate.execute("update c_project_cost_budget_stats set approve = 1 where project_id = " + projectCostBudget.getProjectId());
+                    List<ProjectCostBudget> projectCostBudgetList = jdbcTemplate.query("select project_id from c_project_cost_budget where id in(" + dataId + ")", new BeanPropertyRowMapper<>(ProjectCostBudget.class));
+                    if (projectCostBudgetList.size() > 0) {
+                        projectCostBudgetList.stream().map(ProjectCostBudget::getProjectId).findAny().ifPresent(projectId -> jdbcTemplate.execute("update c_project_cost_budget_stats set approve = 1 where project_id = " + projectId));
                     }
                 } else if (taskStatus == 3) {
                     //已驳回 = approve = 0