From 0b8be6398f4951e652d1443027991816478d3a85 Mon Sep 17 00:00:00 2001
From: "15989082884@163.com" <15989082884@163.com>
Date: Wed, 5 Jul 2023 14:51:04 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E9=A2=84=E7=AE=97IDS?=
 =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E6=9D=A1=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../serve/modules/tbs/controller/TbsBudgetController.java   | 6 ++++++
 .../com/qs/serve/modules/tbs/entity/so/TbsBudgetSo.java     | 3 +++
 2 files changed, 9 insertions(+)

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 390da2d2..eabb0b55 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
@@ -57,6 +57,9 @@ public class TbsBudgetController {
     @PreAuthorize("hasRole('tbs:budget:query')")
     public R<List<TbsBudget>> getList(TbsBudgetSo param){
         List<Long> budgetIds = tbsBudgetMapper.selectBudgetId(param);
+        if(param.getSelectIdList()!=null && param.getSelectIdList().size()>0){
+            budgetIds.addAll(param.getSelectIdList());
+        }
         if(budgetIds.size()>0){
             List<TbsBudget> list = tbsBudgetService.listByIds(budgetIds);
             return R.ok(list);
@@ -73,6 +76,9 @@ public class TbsBudgetController {
     @PreAuthorize("hasRole('tbs:budget:query')")
     public R<PageVo<TbsBudgetVo>> getPage(TbsBudgetSo param){
         List<Long> budgetIds = tbsBudgetMapper.selectBudgetId(param);
+        if(param.getSelectIdList()!=null && param.getSelectIdList().size()>0){
+            budgetIds.addAll(param.getSelectIdList());
+        }
         if(budgetIds.size()>0){
             PageUtil.startPage();
             LambdaQueryWrapper<TbsBudget> lqw = new LambdaQueryWrapper<>();
diff --git a/src/main/java/com/qs/serve/modules/tbs/entity/so/TbsBudgetSo.java b/src/main/java/com/qs/serve/modules/tbs/entity/so/TbsBudgetSo.java
index 6beb84af..7e86d245 100644
--- a/src/main/java/com/qs/serve/modules/tbs/entity/so/TbsBudgetSo.java
+++ b/src/main/java/com/qs/serve/modules/tbs/entity/so/TbsBudgetSo.java
@@ -4,6 +4,7 @@ import java.time.LocalDate;
 import java.io.Serializable;
 import java.math.BigDecimal;
 import java.time.LocalDateTime;
+import java.util.List;
 
 import com.fasterxml.jackson.annotation.JsonFormat;
 import com.fasterxml.jackson.annotation.JsonIgnore;
@@ -23,6 +24,8 @@ import javax.validation.constraints.NotBlank;
 @Data
 public class TbsBudgetSo implements Serializable {
 
+    private List<Long> selectIdList;
+
     /** 预算编码 */
     private String budgetCode;