From faf2b690c4a2bc3e3155c8f1f25635a2d3a3a574 Mon Sep 17 00:00:00 2001
From: "15989082884@163.com" <15989082884@163.com>
Date: Sun, 2 Jul 2023 14:30:29 +0800
Subject: [PATCH] =?UTF-8?q?=E9=A2=84=E8=AE=A1=E5=8F=91=E8=B4=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../tbs/controller/TbsBudgetController.java   | 16 ++++++++++
 .../modules/tbs/entity/bo/TbsBudgetBo.java    |  3 ++
 .../TbsScheduleItemBudgetPreDispatchBo.java   | 29 +++++++++++++++++++
 .../service/impl/TbsBudgetServiceImpl.java    |  1 +
 4 files changed, 49 insertions(+)
 create mode 100644 src/main/java/com/qs/serve/modules/tbs/entity/bo/TbsScheduleItemBudgetPreDispatchBo.java

diff --git a/src/main/java/com/qs/serve/modules/tbs/controller/TbsBudgetController.java b/src/main/java/com/qs/serve/modules/tbs/controller/TbsBudgetController.java
index 2f349cb1..390da2d2 100644
--- a/src/main/java/com/qs/serve/modules/tbs/controller/TbsBudgetController.java
+++ b/src/main/java/com/qs/serve/modules/tbs/controller/TbsBudgetController.java
@@ -164,6 +164,22 @@ public class TbsBudgetController {
         return R.ok();
     }
 
+    /**
+     * 修改预计发货金额
+     * @param param
+     * @return
+     */
+    @PostMapping("/updateScheduleItemPreDispatchAmount")
+    @SysLog(module = SystemModule.Budget, title = "预算", biz = BizType.UPDATE)
+    @PreAuthorize("hasRole('tbs:budget:update')")
+    public R<?> updateScheduleItemPreDispatchAmount(@RequestBody @Valid TbsScheduleItemBudgetPreDispatchBo param){
+        TbsScheduleItemBudget itemBudget = new TbsScheduleItemBudget();
+        itemBudget.setId(param.getId());
+        itemBudget.setPreDispatchAmount(param.getPreDispatchAmount());
+        tbsScheduleItemBudgetService.updateById(itemBudget);
+        return R.ok();
+    }
+
     /**
     * 新增
     * @param param
diff --git a/src/main/java/com/qs/serve/modules/tbs/entity/bo/TbsBudgetBo.java b/src/main/java/com/qs/serve/modules/tbs/entity/bo/TbsBudgetBo.java
index 8000c5cb..7eb426c7 100644
--- a/src/main/java/com/qs/serve/modules/tbs/entity/bo/TbsBudgetBo.java
+++ b/src/main/java/com/qs/serve/modules/tbs/entity/bo/TbsBudgetBo.java
@@ -82,6 +82,9 @@ public class TbsBudgetBo implements Serializable {
 
         /** 预算金额 */
         private BigDecimal amount;
+
+        private BigDecimal preDispatchAmount;
+
     }
 
 }
diff --git a/src/main/java/com/qs/serve/modules/tbs/entity/bo/TbsScheduleItemBudgetPreDispatchBo.java b/src/main/java/com/qs/serve/modules/tbs/entity/bo/TbsScheduleItemBudgetPreDispatchBo.java
new file mode 100644
index 00000000..9ddf1ba7
--- /dev/null
+++ b/src/main/java/com/qs/serve/modules/tbs/entity/bo/TbsScheduleItemBudgetPreDispatchBo.java
@@ -0,0 +1,29 @@
+package com.qs.serve.modules.tbs.entity.bo;
+
+import lombok.Data;
+
+import javax.validation.constraints.DecimalMin;
+import javax.validation.constraints.NotNull;
+import java.io.Serializable;
+import java.math.BigDecimal;
+
+/**
+ * 预算考核期项 Bo
+ * @author YenHex
+ * @since 2022-11-09
+ */
+@Data
+public class TbsScheduleItemBudgetPreDispatchBo implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+    /** 考核期项费用id */
+    private Long id;
+
+    /** 预算金额 */
+    @DecimalMin(value = "0")
+    @NotNull(message = "预计发货金额不能为空")
+    private BigDecimal preDispatchAmount;
+
+}
+
diff --git a/src/main/java/com/qs/serve/modules/tbs/service/impl/TbsBudgetServiceImpl.java b/src/main/java/com/qs/serve/modules/tbs/service/impl/TbsBudgetServiceImpl.java
index ef633575..a5a5fc22 100644
--- a/src/main/java/com/qs/serve/modules/tbs/service/impl/TbsBudgetServiceImpl.java
+++ b/src/main/java/com/qs/serve/modules/tbs/service/impl/TbsBudgetServiceImpl.java
@@ -121,6 +121,7 @@ public class TbsBudgetServiceImpl extends ServiceImpl<TbsBudgetMapper,TbsBudget>
             for (TbsBudgetBo.ScheduleItem boScheduleItem : budgetBo.getScheduleItems()) {
                 if(scheduleItem.getId().equals(boScheduleItem.getScheduleItemId())){
                     itemBudget.setBudgetAmount(boScheduleItem.getAmount());
+                    itemBudget.setPreDispatchAmount(boScheduleItem.getPreDispatchAmount());
                     break;
                 }
             }