From 30d220fd3583067b25447c42abfaa7a94afbecd2 Mon Sep 17 00:00:00 2001
From: Yen <yen1130@qq.com>
Date: Wed, 24 May 2023 11:00:24 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BE=AE=E8=B0=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../modules/bms/controller/BmsChannelController.java     | 9 +++++++++
 .../java/com/qs/serve/modules/bms/entity/BmsChannel.java | 4 ++++
 2 files changed, 13 insertions(+)

diff --git a/src/main/java/com/qs/serve/modules/bms/controller/BmsChannelController.java b/src/main/java/com/qs/serve/modules/bms/controller/BmsChannelController.java
index 39718263..75dec58b 100644
--- a/src/main/java/com/qs/serve/modules/bms/controller/BmsChannelController.java
+++ b/src/main/java/com/qs/serve/modules/bms/controller/BmsChannelController.java
@@ -14,6 +14,7 @@ import com.qs.serve.modules.bms.controller.my.BmsChannelMyController;
 import com.qs.serve.modules.bms.entity.BmsChannelPoint;
 import com.qs.serve.modules.bms.entity.BmsMasterUser;
 import com.qs.serve.modules.bms.entity.BmsSupplierVisitType;
+import com.qs.serve.modules.bms.entity.bo.BmsSupplierChannelBo;
 import com.qs.serve.modules.bms.mapper.BmsMasterUserMapper;
 import com.qs.serve.modules.bms.service.BmsChannelPointService;
 import com.qs.serve.modules.bms.service.BmsSupplierChannelService;
@@ -30,6 +31,7 @@ import com.qs.serve.modules.bms.entity.BmsChannel;
 import com.qs.serve.modules.bms.service.BmsChannelService;
 
 import javax.validation.Valid;
+import java.util.Arrays;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -51,6 +53,7 @@ public class BmsChannelController {
     private BmsMasterUserMapper bmsMasterUserMapper;
     private BmsSupplierChannelService bmsSupplierChannelService;
     private BmsChannelMyController bmsChannelMyController;
+    private BmsSupplierChannelController bmsSupplierChannelController;
 
     /**
      * 导出列表
@@ -153,6 +156,12 @@ public class BmsChannelController {
     public R<?> save(@RequestBody @Valid BmsChannel param){
         param.setChannelCode("D"+ StringUtils.genShortId());
         boolean result = bmsChannelService.save(param);
+        if(result&&CollectionUtil.isNotEmpty(param.getSupplierIds())){
+            BmsSupplierChannelBo channelBo = new BmsSupplierChannelBo();
+            channelBo.setChannelIds(Arrays.asList(param.getId()));
+            channelBo.setSupplierIds(param.getSupplierIds());
+            bmsSupplierChannelController.save(channelBo);
+        }
         return R.isTrue(result);
     }
 
diff --git a/src/main/java/com/qs/serve/modules/bms/entity/BmsChannel.java b/src/main/java/com/qs/serve/modules/bms/entity/BmsChannel.java
index 5fa04652..5ea036eb 100644
--- a/src/main/java/com/qs/serve/modules/bms/entity/BmsChannel.java
+++ b/src/main/java/com/qs/serve/modules/bms/entity/BmsChannel.java
@@ -100,5 +100,9 @@ public class BmsChannel implements Serializable {
     @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss",timezone = "GMT+8")
     private LocalDateTime selectEndTime;
 
+    /** 选择的供应商ID,用于新增 */
+    @TableField(exist = false)
+    private List<Long> supplierIds;
+
 }