From 7792a1396f3e48e4076346a0f4d7372b1415da07 Mon Sep 17 00:00:00 2001
From: OathK1per <zhangypg@gmail.com>
Date: Tue, 13 Dec 2022 16:41:19 +0800
Subject: [PATCH] =?UTF-8?q?=E6=A6=82=E7=AE=97=E5=AE=A1=E6=A0=B8=E5=A4=84?=
 =?UTF-8?q?=E6=96=B0=E5=A2=9E=E9=A1=B9=E7=9B=AE=E9=A2=84=E7=AE=97=E7=9A=84?=
 =?UTF-8?q?=E4=BF=9D=E5=AD=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/cn/palmte/work/service/ProjectEstimateService.java  | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/main/java/cn/palmte/work/service/ProjectEstimateService.java b/src/main/java/cn/palmte/work/service/ProjectEstimateService.java
index 9b1e66a..2ca5694 100644
--- a/src/main/java/cn/palmte/work/service/ProjectEstimateService.java
+++ b/src/main/java/cn/palmte/work/service/ProjectEstimateService.java
@@ -264,6 +264,8 @@ public class ProjectEstimateService {
 
         //projectInstanceService.startEstimateProcessInstance(p, admin);
 
+        projectBudgetSave(p);
+
         estimateApprove(0, p, admin);
 
         return project;
@@ -302,6 +304,9 @@ public class ProjectEstimateService {
 
     private void projectBudgetSave(Project estimateProject) {
         ProjectBudget projectBudget = new ProjectBudget();
+        if (projectBudgetRepository.findFirstByProjectId(estimateProject.getId()) != null) {
+            projectBudget = projectBudgetRepository.findFirstByProjectId(estimateProject.getId());
+        }
         projectBudget.setProjectId(estimateProject.getId());
         projectBudget.setProjectNoBudget(estimateProject.getProjectNo());
         projectBudget.setNameBudget(estimateProject.getName());