导出

platform-admin/src/main/java/com/platform/service/impl/TbCfUserInfoServiceImpl.java

@@ -13,17 +13,11 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
 
-import java.io.FileNotFoundException;
-import java.io.FileOutputStream;
-import java.io.IOException;
-import java.io.OutputStream;
+import java.io.*;
 import java.math.BigDecimal;
 import java.sql.Timestamp;
 import java.text.DecimalFormat;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * 用户表Service实现类
@@ -155,6 +149,7 @@ public class TbCfUserInfoServiceImpl implements TbCfUserInfoService {
     @Override
     public void importData(MultipartFile multipartFile) {
         List<String[]> excelData = ExcelImport.getExcelData(multipartFile);
+        excelData.remove(0);
         for (int i = 0; i < excelData.size(); i++) {
             String[] stringData = excelData.get(i);
             TbCfUserInfoEntity user = new TbCfUserInfoEntity();
@@ -203,7 +198,8 @@ public class TbCfUserInfoServiceImpl implements TbCfUserInfoService {
 
         OutputStream fis;
         try {
-            fis = new FileOutputStream("D:\\virtual\\virtualUsers.xlsx");
+            String filename = encodingFilename("virtualUsers");
+            fis = new FileOutputStream(getAbsoluteFile(filename));
 
             ee1.getWorkbook().write(fis);
 
@@ -215,5 +211,25 @@ public class TbCfUserInfoServiceImpl implements TbCfUserInfoService {
 
     }
 
+    /**
+     * 编码文件名
+     */
+    public String encodingFilename(String filename) {
+        filename = UUID.randomUUID().toString() + "_" + filename + ".xlsx";
+        return filename;
+    }
 
+    /**
+     * 获取下载路径
+     *
+     * @param filename 文件名称
+     */
+    public String getAbsoluteFile(String filename) {
+        String downloadPath = "D:\\virtual\\" + filename;
+        File desc = new File(downloadPath);
+        if (!desc.getParentFile().exists()) {
+            desc.getParentFile().mkdirs();
+        }
+        return downloadPath;
+    }
 }