微调

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);
     }
 

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;
+
 }