From b9903ead016b8b1aa68eb04b48fca3b53fdab0d3 Mon Sep 17 00:00:00 2001
From: cloudroam <cloudroam>
Date: 星期一, 30 十二月 2024 10:42:43 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master-v4' into master-v4

---
 src/main/java/com/mzl/flower/service/impl/sms/SmsTaskServiceImpl.java |  131 ++++++++++++++++++++++++++++++++++---------
 1 files changed, 102 insertions(+), 29 deletions(-)

diff --git a/src/main/java/com/mzl/flower/service/impl/sms/SmsTaskServiceImpl.java b/src/main/java/com/mzl/flower/service/impl/sms/SmsTaskServiceImpl.java
index e7c3523..f3af2d8 100644
--- a/src/main/java/com/mzl/flower/service/impl/sms/SmsTaskServiceImpl.java
+++ b/src/main/java/com/mzl/flower/service/impl/sms/SmsTaskServiceImpl.java
@@ -8,6 +8,8 @@
 import com.aliyuncs.dysmsapi.model.v20170525.SendSmsResponse;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
 import com.mzl.flower.config.OssProperties;
 import com.mzl.flower.config.exception.ValidationException;
 import com.mzl.flower.config.security.SecurityUtils;
@@ -35,6 +37,7 @@
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.CollectionUtils;
+import org.springframework.util.ObjectUtils;
 import org.springframework.util.StringUtils;
 
 import java.io.File;
@@ -83,7 +86,7 @@
     }
 
     @Override
-    public void saveSmsTask(SmsTaskDTO smsTaskDTO) {
+    public void saveSmsTask(SmsTaskDTO smsTaskDTO) throws IOException {
         //校验
         if (StringUtils.isEmpty(smsTaskDTO.getName())) {
             throw new ValidationException("短信名称不能为空");
@@ -97,7 +100,7 @@
             throw new ValidationException("接收号码类型不能为空");
         }
 
-        if ((Constants.SMS_RECEIVE_TYPE.INPUT.name().equals(smsTaskDTO.getType()) || Constants.SMS_RECEIVE_TYPE.SELECT.name().equals(smsTaskDTO.getType())) && StringUtils.isEmpty(smsTaskDTO.getPhones())) {
+        if (Constants.SMS_RECEIVE_TYPE.INPUT.name().equals(smsTaskDTO.getType()) && StringUtils.isEmpty(smsTaskDTO.getPhones())) {
             throw new ValidationException("手机号不能为空");
         }
 
@@ -105,7 +108,7 @@
             throw new ValidationException("导入文件不能为空");
         }
 
-        if (Constants.SMS_RECEIVE_TYPE.INPUT.name().equals(smsTaskDTO.getType()) || Constants.SMS_RECEIVE_TYPE.SELECT.name().equals(smsTaskDTO.getType())) {
+        if (Constants.SMS_RECEIVE_TYPE.INPUT.name().equals(smsTaskDTO.getType()) ) {
 
             //解析手机号,包含不同平台的换行符
             String text = smsTaskDTO.getPhones();
@@ -127,26 +130,28 @@
             dealImportExcel(smsTaskDTO);
         }
         List<SmsUserDTO> smsUserDTOS = smsTaskDTO.getSmsUserDTOS();
-        List<String> userIds = smsUserDTOS.stream()
-                .map(SmsUserDTO::getUserId)
-                .collect(Collectors.toList());
-
-        List<String> userPhones = smsUserDTOS.stream()
-                .map(SmsUserDTO::getUserPhone)
-                .collect(Collectors.toList());
-
-        if(CollectionUtils.isEmpty(userPhones)){
-            String phones = userPhones.stream()
-                    .map(Object::toString) // 确保每个元素都转换为字符串
-                    .collect(Collectors.joining("\n")); // 使用换行符连接字符串
-            smsTaskDTO.setPhones(phones); // 假设有一个setPhones方法用于设置phones字段
-        }
         SmsTaskDO smsTaskDO = new SmsTaskDO();
         BeanUtils.copyProperties(smsTaskDTO, smsTaskDO);
-        if (CollectionUtils.isEmpty(userIds)) {
-            String userIdInfos = userIds.stream().map(Object::toString) // 确保每个元素都转换为字符串
-                    .collect(Collectors.joining(";")); // 使用换行符连接字符串
-            smsTaskDO.setUserIds(userIdInfos);
+        if(!CollectionUtils.isEmpty(smsUserDTOS)){
+            List<String> userIds = smsUserDTOS.stream()
+                    .map(SmsUserDTO::getUserId)
+                    .collect(Collectors.toList());
+
+            List<String> userPhones = smsUserDTOS.stream()
+                    .map(SmsUserDTO::getUserPhone)
+                    .collect(Collectors.toList());
+            if(!CollectionUtils.isEmpty(userPhones)){
+                String phones = userPhones.stream()
+                        .map(Object::toString) // 确保每个元素都转换为字符串
+                        .collect(Collectors.joining("\n")); // 使用换行符连接字符串
+                smsTaskDTO.setPhones(phones); // 假设有一个setPhones方法用于设置phones字段
+            }
+            if (!CollectionUtils.isEmpty(userIds)) {
+                String userIdInfos = userIds.stream().map(Object::toString) // 确保每个元素都转换为字符串
+                        .collect(Collectors.joining(";")); // 使用换行符连接字符串
+                smsTaskDO.setUserIds(userIdInfos);
+                smsTaskDTO.setNum((long) userIds.size());
+            }
         }
         smsTaskDO.setStatus(Constants.SMS_TASK_STATUS.wait_publish.name());
         smsTaskDO.setPhones(smsTaskDTO.getPhones());
@@ -154,7 +159,22 @@
         smsTaskMapper.insert(smsTaskDO);
     }
 
-    private void dealImportExcel(SmsTaskDTO smsTaskDTO) {
+    private void dealImportExcel(SmsTaskDTO smsTaskDTO) throws IOException {
+        String fileUrlMessage = "";
+        // 创建ObjectMapper实例
+        ObjectMapper objectMapper = new ObjectMapper();
+        JsonNode rootNode = objectMapper.readTree(smsTaskDTO.getFileUrl());
+        if (rootNode.isArray()) {
+            JsonNode firstElement = rootNode.get(0);
+            if (firstElement.has("url")) {
+                 fileUrlMessage = firstElement.get("url").asText();
+            } else {
+                throw new ValidationException("URL字段不存在");
+            }
+        } else {
+            throw new ValidationException("JSON数组为空或不是数组");
+        }
+
         String endPoint = ossProperties.getEndpoint();
         String accessKeyId = ossProperties.getKeyid();
         String accessKeySecret = ossProperties.getKeysecret();
@@ -164,7 +184,7 @@
         try {
             // 下载Excel文件到本地临时文件
             File tempFile = File.createTempFile("temp", ".xlsx");
-            String fileUrl = smsTaskDTO.getFileUrl();
+            String fileUrl = fileUrlMessage;
             String objectKey = fileUrl.replaceFirst("^https?://[^/]+/", ""); // 去掉协议部分
             ossClient.getObject(new GetObjectRequest(bucketName, objectKey), tempFile);
 
@@ -235,7 +255,7 @@
     }
 
     @Override
-    public void updateSmsTask(SmsTaskDTO smsTaskDTO) {
+    public void updateSmsTask(SmsTaskDTO smsTaskDTO) throws IOException {
         SmsTaskDO smsTaskDO = smsTaskMapper.selectById(smsTaskDTO.getId());
         if (!smsTaskDO.getStatus().equals(Constants.SMS_TASK_STATUS.wait_publish.name())) {
             throw new ValidationException("非待发布的任务不可编辑");
@@ -253,7 +273,7 @@
             throw new ValidationException("接收号码类型不能为空");
         }
 
-        if ((Constants.SMS_RECEIVE_TYPE.INPUT.name().equals(smsTaskDTO.getType()) || Constants.SMS_RECEIVE_TYPE.SELECT.name().equals(smsTaskDTO.getType())) && StringUtils.isEmpty(smsTaskDTO.getPhones())) {
+        if (Constants.SMS_RECEIVE_TYPE.INPUT.name().equals(smsTaskDTO.getType()) && StringUtils.isEmpty(smsTaskDTO.getPhones())) {
             throw new ValidationException("手机号不能为空");
         }
 
@@ -261,11 +281,51 @@
             throw new ValidationException("导入文件不能为空");
         }
 
-        if(!smsTaskDTO.getFileUrl().equals(smsTaskDO.getFileUrl())){
-            dealImportExcel(smsTaskDTO);
+        if (Constants.SMS_RECEIVE_TYPE.INPUT.name().equals(smsTaskDTO.getType()) ) {
+
+            //解析手机号,包含不同平台的换行符
+            String text = smsTaskDTO.getPhones();
+
+            // 使用正则表达式匹配所有类型的换行符
+            String[] lines = text.split("\\r?\\n|\\r");
+
+            // 将数组转换为 List
+            List<String> lineList = Arrays.asList(lines);
+            lineList.forEach(l -> {
+                boolean validPhoneNumber = isValidPhoneNumber(l);
+                if (!validPhoneNumber) {
+                    throw new ValidationException(l + "不是合法的手机号");
+                }
+            });
+            smsTaskDTO.setNum((long) lineList.size());
         }
 
-        BeanUtils.copyProperties(smsTaskDTO, smsTaskDO);
+        if(Constants.SMS_RECEIVE_TYPE.IMPORT.name().equals(smsTaskDTO.getType()) && !smsTaskDTO.getFileUrl().equals(smsTaskDO.getFileUrl())){
+            dealImportExcel(smsTaskDTO);
+        }
+        List<SmsUserDTO> smsUserDTOS = smsTaskDTO.getSmsUserDTOS();
+        if(!CollectionUtils.isEmpty(smsUserDTOS)){
+            List<String> userIds = smsUserDTOS.stream()
+                    .map(SmsUserDTO::getUserId)
+                    .collect(Collectors.toList());
+
+            List<String> userPhones = smsUserDTOS.stream()
+                    .map(SmsUserDTO::getUserPhone)
+                    .collect(Collectors.toList());
+            if(!CollectionUtils.isEmpty(userPhones)){
+                String phones = userPhones.stream()
+                        .map(Object::toString) // 确保每个元素都转换为字符串
+                        .collect(Collectors.joining("\n")); // 使用换行符连接字符串
+                smsTaskDTO.setPhones(phones); // 假设有一个setPhones方法用于设置phones字段
+            }
+            if (!CollectionUtils.isEmpty(userIds)) {
+                String userIdInfos = userIds.stream().map(Object::toString) // 确保每个元素都转换为字符串
+                        .collect(Collectors.joining(";")); // 使用换行符连接字符串
+                smsTaskDO.setUserIds(userIdInfos);
+                smsTaskDTO.setNum((long) userIds.size());
+            }
+        }
+        BeanUtils.copyProperties(smsTaskDTO, smsTaskDO,"userIds");
         smsTaskDO.update(SecurityUtils.getUserId());
         smsTaskDO.setPhones(smsTaskDTO.getPhones());
         smsTaskMapper.updateById(smsTaskDO);
@@ -318,7 +378,7 @@
         });
     }
 
-    @Override
+    @Override 
     public List<SmsSelectVO> getSelectList(Long id) {
         List<SmsSelectVO> smsSelectVOList = null;
         SmsTaskDO smsTaskDO = smsTaskMapper.selectById(id);
@@ -332,6 +392,19 @@
         }
     }
 
+    @Override
+    public SmsTaskVO getDetailById(Long id) {
+        SmsTaskDO smsTaskDO = smsTaskMapper.selectById(id);
+        SmsTaskVO smsTaskVO=new SmsTaskVO();
+        BeanUtils.copyProperties(smsTaskDO,smsTaskVO);
+        if(!ObjectUtils.isEmpty(smsTaskDO)){
+            List<SmsSelectVO> selectList = getSelectList(id);
+            smsTaskVO.setSmsUserDTOS(selectList);
+            return smsTaskVO;
+        }
+        return null;
+    }
+
     private List<SmsTaskDetailDO> createSmsTaskDetails(SmsTaskDO smsTaskDO, List<String> phoneNumbers) {
         return phoneNumbers.stream().map(phone -> {
             SmsTaskDetailDO detail = new SmsTaskDetailDO();

--
Gitblit v1.9.3