Merge remote-tracking branch 'origin/master'

service-manager/src/main/java/com/winsun/controller/RhPhoneController.java

@@ -8,16 +8,13 @@ import com.winsun.auth.core.base.controller.BaseController;
 import com.winsun.auth.core.common.model.ResponseData;
 import com.winsun.auth.core.common.model.ResponseData;
 import com.winsun.bean.RhPhone;
 import com.winsun.bean.RhPhone;
 import com.winsun.mapper.RhPhoneMapper;
 import com.winsun.mapper.RhPhoneMapper;
+import com.winsun.utils.ImportExcel;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RequestMethod;
-import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
 
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 
 /**
 /**
  * @Author: chenhaojian
  * @Author: chenhaojian
@@ -29,6 +26,12 @@ public class RhPhoneController extends BaseController {
 
 
     private static RhPhoneMapper rhPhoneMapper;
     private static RhPhoneMapper rhPhoneMapper;
 
 
+    /**
+     * 导入
+     */
+    private static final int HEADER_NUM = -1;
+    private static final int SHEET_INDEX = 0;
+
     @Autowired
     @Autowired
     public RhPhoneController(RhPhoneMapper rhPhoneMapper){
     public RhPhoneController(RhPhoneMapper rhPhoneMapper){
         RhPhoneController.rhPhoneMapper = rhPhoneMapper;
         RhPhoneController.rhPhoneMapper = rhPhoneMapper;
@@ -87,4 +90,60 @@ public class RhPhoneController extends BaseController {
         }
         }
         return ResponseData.success();
         return ResponseData.success();
     }
     }
+
+
+
+    /**
+     * 导入融合号码数据
+     * @return
+     */
+
+    @PostMapping(value = "importAccNbr")
+    public ResponseData<String> importAccNbr(@RequestParam(name = "file") MultipartFile file){
+        StringBuffer error = new StringBuffer();
+        if (!StringUtils.endsWithAny(file.getOriginalFilename(), "xlsx", "xls")) {
+            return ResponseData.error("手工上传数据仅支持Excel文件，其他格式不支持！");
+        }
+        if ( (double)file.getSize()/1048576 > 100) {
+            return ResponseData.error("文件过大，无法上传");
+        }
+
+        ImportExcel importExcel = null;
+        try {
+            importExcel = new ImportExcel(file, HEADER_NUM, SHEET_INDEX);
+        } catch (Exception e) {
+            return ResponseData.error("导入异常！");
+        }
+
+
+        List<Map<String, Object>> listMap = importExcel.getDataListMap();
+        System.out.println(listMap);
+        if (listMap.size() == 1) {
+            return ResponseData.error("文件不能为空");
+        }
+        //检查excel表中是否有重复号码
+        int size = listMap.size();
+        Set<String> accountSet = new HashSet<>(size);
+        for (int i = 0; i < size; i++) {
+            Map<String, Object> temp = listMap.get(i);
+            accountSet.add((String) temp.get("a"));
+        }
+        if (accountSet.size() != size) {
+            return ResponseData.error("文件中有重复的订单编号，请确认订单编号不能重复!");
+        }
+        try {
+            for (int i = 0; i < listMap.size(); i++) {
+                RhPhone rhPhone=new RhPhone();
+                Map<String, Object> map = listMap.get(i);
+                rhPhone.setPhone(map.get("a").toString());
+                rhPhoneMapper.insert(rhPhone);
+            }
+            return ResponseData.success(error.toString(), "导入完成");
+        }catch (Exception e){
+            e.printStackTrace();
+            return ResponseData.error("导入异常！");
+        }
+
+    }
+
 }
 }