导入相关代码优化

nafmii-admin/src/main/java/org/nafmii/admin/user/controller/ImportExcelController.java

@@ -25,9 +25,8 @@ public class ImportExcelController {
     @ApiOperation( value = "中心端批量导入Excel",notes = "中心端批量导入Excel")
     @PostMapping("/batchImportExcel")
     @ResponseBody
-    public ApiResponse batchImportExcel(MultipartFile file) throws Exception{
-        List<ImportExcelDto> list = EasyExcel.read(file.getInputStream()).sheet(0).head(ImportExcelDto.class).doReadSync();
-        ApiResponse result = iImportExcelService.batchImportExcel(list);
+    public ApiResponse batchImportExcel(MultipartFile file){
+        ApiResponse result = iImportExcelService.batchImportExcel(file);
         return result;
     }
 }

nafmii-admin/src/main/java/org/nafmii/admin/user/dao/ImportExcelMapper.java

@@ -25,5 +25,5 @@ public interface ImportExcelMapper extends BaseMapper<CenterUserPO> {
      * @version:
      * @remark:
      */
-    int insertExcelData(List<ImportExcelDto> list);
+    int insertExcelData(ImportExcelDto data);
 }

nafmii-admin/src/main/java/org/nafmii/admin/user/service/IImportExcelService.java

@@ -2,6 +2,7 @@ package org.nafmii.admin.user.service;
 
 import org.nafmii.admin.user.dto.ImportExcelDto;
 import org.nafmii.common.response.ApiResponse;
+import org.springframework.web.multipart.MultipartFile;
 
 import java.util.List;
 
@@ -22,6 +23,6 @@ public interface IImportExcelService {
      * @version:
      * @remark:
      */
-    ApiResponse batchImportExcel(List<ImportExcelDto> list);
+    ApiResponse batchImportExcel(MultipartFile file);
 
 }

nafmii-admin/src/main/java/org/nafmii/admin/user/service/impl/ImportExcelServiceImpl.java

 
 package org.nafmii.admin.user.service.impl;
 
+import com.alibaba.excel.EasyExcel;
 import lombok.extern.slf4j.Slf4j;
 import org.nafmii.admin.user.dao.ImportExcelMapper;
 import org.nafmii.admin.user.dto.ImportExcelDto;
@@ -18,7 +19,9 @@ import org.nafmii.admin.user.service.IImportExcelService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import org.springframework.web.multipart.MultipartFile;
 
+import java.io.IOException;
 import java.io.InputStream;
 import java.util.ArrayList;
 import java.util.List;
@@ -42,13 +45,25 @@ public class ImportExcelServiceImpl implements IImportExcelService {
     private ImportExcelMapper importExcelMapper;
 
     @Override
-    public ApiResponse batchImportExcel(List<ImportExcelDto> list) {
-        int i = importExcelMapper.insertExcelData(list);
-        if (i < 0) {
-            return new ApiResponse(ApiResponse.FAIL, ApiResponse.FAIL_TEXT);
-        } else {
-            return new ApiResponse(ApiResponse.SUCCESS, ApiResponse.SUCCESS_TEXT);
+    public ApiResponse batchImportExcel(MultipartFile file) {
+        List<ImportExcelDto> list = null;
+        try {
+            list = EasyExcel.read(file.getInputStream()).sheet(0).head(ImportExcelDto.class).doReadSync();
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        log.info("导入后生成的数据源=============>"+list);
+        if (!list.isEmpty()) {
+            for (ImportExcelDto data : list) {
+                int i = importExcelMapper.insertExcelData(data);
+                if (i < 0) {
+                    return new ApiResponse(ApiResponse.FAIL, ApiResponse.FAIL_TEXT);
+                } else {
+                    return new ApiResponse(ApiResponse.SUCCESS, ApiResponse.SUCCESS_TEXT);
+                }
+            }
         }
+        return new ApiResponse(ApiResponse.FAIL, "导入时数据为空，导入失败");
     }
 
 

nafmii-admin/src/main/resources/mapper/ImportExcelMapper.xml

@@ -10,15 +10,12 @@
         <result column="COLUMN3" property="column3" />
     </resultMap>
 
-    <insert id ="insertExcelData" parameterType="java.util.List" >
+    <insert id ="insertExcelData" parameterType="org.nafmii.admin.user.dto.ImportExcelDto" >
         insert into IMPORT_EXCEL_TEST
         (COLUMN1,COLUMN2,COLUMN3)
-        values
-        <foreach collection ="list" item="ImportExcelPO" index= "index" separator =",">
-            (
-            #{ImportExcelPO.column1}, #{ImportExcelPO.column2},#{ImportExcelPO.column3}
-            )
-        </foreach >
+        values(#{ImportExcelPO.column1},
+        #{ImportExcelPO.column2},
+        #{ImportExcelPO.column3})
     </insert >
 
 </mapper>