完整版excel上傳導入讀寫批量數據並將反饋結果寫入遠程exel中

思路：excel的讀寫借助於poi框架，在寫入遠程的時候，是不能直接寫入的，本博主將傳入的文件再次拉下來寫到項目臨時文件中，然后，在臨時文件中寫入，然后，以同樣的名稱路徑覆蓋掉遠程的就可以了，稍微有點繞了，從遠端獲取文件，需要通過流來寫到項目臨時文件中，具體見下方代碼，代碼中有部分業務刪減，該代碼是在工作中運行通過的。模板是我們自己制定的，所以只適合已有模板。

文件工具類及方法：

package com.XXXX.XXXXX.utils;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.io.FileUtils;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.ResourceUtils;

import com.sun.javafx.scene.control.skin.TreeTableRowSkin;
import com.xxx.cloud.common.utils.DateFormatUtil;
import com.xxx.cloud.common.utils.StringUtil;

public class ReadExcelUtil {
    
    private static final String EXCEL_XLS = ".xls";
    private static final String EXCEL_XLSX = ".xlsx";
    
    /**
     *讀取excel數據
     * @throws Exception 
     *
     */
    public static List<List<String>> readExcelInfo(String url) throws Exception{
//        支持excel2003、2007
        File excelFile = new File(url);//創建excel文件對象
        InputStream is = new FileInputStream(excelFile);//創建輸入流對象
        checkExcelVaild(excelFile);
        Workbook workbook = getWorkBook(is, excelFile);
//        Workbook workbook = WorkbookFactory.create(is);//同時支持2003、2007、2010
//        獲取Sheet數量
        int sheetNum = workbook.getNumberOfSheets();
        sheetNum = 1;//限制模板只在一個工作簿上操作
//      創建二維數組保存所有讀取到的行列數據，外層存行數據，內層存單元格數據
        List<List<String>> dataList = new ArrayList<List<String>>();
//        遍歷工作簿中的sheet,第一層循環所有sheet表
        for(int index = 0;index<sheetNum;index++){
            Sheet sheet = workbook.getSheetAt(index);
            if(sheet==null){
                continue;
            }
//            如果當前行沒有數據跳出循環，第二層循環單sheet表中所有行
            for(int rowIndex=0;rowIndex<=sheet.getLastRowNum();rowIndex++){
                System.out.println(sheet.getLastRowNum()+"====");
                Row row = sheet.getRow(rowIndex);
                if(row==null){
                    continue;
                }
//                遍歷每一行的每一列，第三層循環行中所有單元格
                List<String> cellList = new ArrayList<String>();
                for(int cellIndex=0;cellIndex<row.getLastCellNum();cellIndex++){
                    Cell cell = row.getCell(cellIndex);
                    System.out.println(cellIndex);
                    cellList.add(getCellValue(cell));
                }
                dataList.add(cellList);
            }
            
        }
        is.close();
        return dataList;
        }
    /**
     *獲取單元格的數據,暫時不支持公式
     * 
     *
     */
        public static String getCellValue(Cell cell){
            CellType cellType = cell.getCellTypeEnum();
            if(cellType==null){
                return null;
            }
            String cellValue = "";
            if(cell==null || cell.toString().trim().equals("")){
                return null;
            }
            
            if(cellType==CellType.STRING){
                cellValue = cell.getStringCellValue().trim();
                return cellValue = StringUtil.isEmpty(cellValue)?"":cellValue;
            }
            if(cellType==CellType.NUMERIC){
                if (HSSFDateUtil.isCellDateFormatted(cell)) {  //判斷日期類型
                     cellValue = DateFormatUtil.formatDurationYMD(cell.getDateCellValue().getTime());
                 } else {  //否
                     cellValue = new DecimalFormat("#.######").format(cell.getNumericCellValue()); 
                 } 
                return cellValue;
            }
            if(cellType==CellType.BOOLEAN){
                cellValue = String.valueOf(cell.getBooleanCellValue());
                return cellValue;
            }
            return null;
            
        }
    /**
     *判斷excel的版本，並根據文件流數據獲取workbook
     * @throws IOException 
     *
     */
    public static Workbook getWorkBook(InputStream is,File file) throws Exception{
        
        Workbook workbook = null;
        if(file.getName().endsWith(EXCEL_XLS)){
            workbook = new HSSFWorkbook(is);
        }else if(file.getName().endsWith(EXCEL_XLSX)){
            workbook = new XSSFWorkbook(is);
        }
        
        return workbook;
    }
    /**
     *校驗文件是否為excel
     * @throws Exception 
     * 
     *
     */
    public static void checkExcelVaild(File file) throws Exception {
        String message = "該文件是EXCEL文件！";
        if(!file.exists()){
            message = "文件不存在！";
            throw new Exception(message);
        }
        if(!file.isFile()||((!file.getName().endsWith(EXCEL_XLS)&&!file.getName().endsWith(EXCEL_XLSX)))){
            message = "文件不是Excel";
            throw new Exception(message);
        }
    }
    /**
     *校驗上傳的excel模板是否正確
     * 
     * 
     *
     */
    public static boolean checkExcelTemplate(String url){
            try {
                List<List<String>> list = ReadExcelUtil.readExcelInfo(url);
                for(int i=0;i<list.size();i++){
                    if(i==0){
                        if(!list.get(i).get(0).trim().equals("公司ID")||!list.get(i).get(1).trim().equals("產品ID")||!list.get(0).get(2).trim().equals("設備類型")
                                ||!list.get(i).get(3).trim().equals("設備型號")||!list.get(i).get(4).trim().equals("設備名稱")){
                            return false;
                        }
                    }
                    if(i==2){
                        if(!list.get(i).get(0).trim().equals("設備ID")||!list.get(i).get(1).trim().equals("結果")){
                            return false;
                        }
                    }
            } 
                }catch (Exception e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            return true;

        }
    /**
     * 將反饋結果寫入excel中
     * */
    public static void writeExcelResult(String url,List<Integer> result) throws Exception{
//        支持excel2003、2007
        File excelFile = new File(url);//創建excel文件對象
        InputStream is = new FileInputStream(excelFile);//創建輸入流對象
        checkExcelVaild(excelFile);
        Workbook workbook = getWorkBook(is, excelFile);
//        Workbook workbook = WorkbookFactory.create(is);//同時支持2003、2007、2010
//        獲取Sheet數量
        int sheetNum = workbook.getNumberOfSheets();
        sheetNum = 1;//限制模板只在一個工作簿上操作
//        遍歷工作簿中的sheet,第一層循環所有sheet表
        for(int index = 0;index<sheetNum;index++){
            Sheet sheet = workbook.getSheetAt(index);
            if(sheet==null){
                continue;
            }
//            如果當前行沒有數據跳出循環，第二層循環單sheet表中所有行
            for(int rowIndex=3;rowIndex<=sheet.getLastRowNum();rowIndex++){
                Row row = sheet.getRow(rowIndex);
                row.createCell(1).setCellValue(result.get(rowIndex-3));
            }
            
        }
        FileOutputStream outputStream = new FileOutputStream(url);
        workbook.write(outputStream);
        outputStream.close();
        }
      /** 
     * 根據地址獲得客戶上傳的excel字節流 
     * @param fileUrl 網絡連接地址 
     * @return 
     */  
    public static byte[] getExcelFromAliyun(String fileUrl){  
        try {  
            URL url = new URL(fileUrl);  
            HttpURLConnection conn = (HttpURLConnection)url.openConnection();  
            conn.setRequestMethod("GET");  
            conn.setConnectTimeout(5 * 1000);  
            InputStream inStream = conn.getInputStream();//通過輸入流獲取excelFile數據  
            byte[] excelFile = readInputStream(inStream);//得到excelFile的二進制數據  
            return excelFile;  
        } catch (Exception e) {  
            e.printStackTrace();  
        }  
        return null;  
    }  
    /** 
     * 從網上得到的輸入流中獲取數據轉換為二進制數據 
     * @param inStream 輸入流 
     * @return 
     * @throws Exception 
     */  
    public static byte[] readInputStream(InputStream inStream) throws Exception{  
        ByteArrayOutputStream outStream = new ByteArrayOutputStream();  
        byte[] buffer = new byte[1024];  
        int len = 0;  
        while( (len=inStream.read(buffer)) != -1 ){  
            outStream.write(buffer, 0, len);  
        }  
        inStream.close();  
        return outStream.toByteArray();  
    }
    /** 
     * 將文件寫入到目標目錄中 
     * @param excel 文件數據流
     * @param fileName 文件保存時的名稱 
     */  
    public static void writeFileToDest(byte[] excelFile, File dest){  
        try {  
            FileOutputStream out = new FileOutputStream(dest);  
            out.write(excelFile);  
            out.flush();  
            out.close();  
        } catch (Exception e) {  
            e.printStackTrace();  
        }  
    }
    /***
     * 
     * 在項目中創建臨時文件
     * @throws IOException 
     * */
    public static File createTempFile(String fileName) throws IOException{
        File path = new File(ResourceUtils.getURL("classpath:").getPath());
        if(!path.exists()) path = new File("");
        File upload = new File(path.getAbsolutePath(),"static/images/upload/");
        if(!upload.exists()) upload.mkdirs();
        File tempFile = new File(upload+"/"+fileName);
        if(!tempFile.getParentFile().exists()){
            tempFile.getParentFile().mkdirs();//創建父級文件路徑
            tempFile.createNewFile();//創建文件
        }
        return tempFile;
    }
}

contrller層代碼

@PostMapping("/addBatchDevice")
    public ResponseObj addBatchDevice(@RequestBody JSONObject obj) throws IOException{
        logger.info("導入批量設備："+obj.toJSONString());
        ResponseObj response = new ResponseObj();
        String id = obj.getString("id");
        BatchRecord batchRecord = deviceService.selectBatchRecordById(id);
        String path = aliConstants.aliyunHostOuter+"/"+batchRecord.getFilePath();
//        將該該文件下載出來保存到項目中
        byte[] excelFile = ReadExcelUtil.getExcelFromAliyun(path);
        File tempFile = ReadExcelUtil.createTempFile(batchRecord.getFileName());
        ReadExcelUtil.writeFileToDest(excelFile, tempFile);
        String url = tempFile.getAbsolutePath();
        String companyId = null;
        String productId = null;
        Integer deviceType = null;
        String model = null;
        String deviceName = null;
        boolean flag = ReadExcelUtil.checkExcelTemplate(url);
        if(!flag){
            response.setData(Defined.STATUS_ERROR);
            response.setMessage("文件有誤，請根據模板上傳文件。");
            return response;
        }
        List<Integer> result = new ArrayList<Integer>();//存放反饋信息 
        try {
            List<List<String>> list = ReadExcelUtil.readExcelInfo(url);
            for(int i=0;i<list.size();i++){
            if(i==0||i==2){
                continue;//跳過模板第1,3行
            }    
            if(i==1){
                companyId = list.get(i).get(0);
                productId = list.get(i).get(1);
                deviceType = Integer.valueOf(list.get(i).get(2));
                model = list.get(i).get(3);
                deviceName = list.get(i).get(4);
            }
            if(i>2){
//                new一個對象按照相應的字段設置進去就可以了，這里省略對象設置值，字段如下：
                Device device = new Device();
                String deviceId = IdGen.uuid();
                device.setId(deviceId);
                //省略部分業務代碼
                DeviceFields deviceFields = new DeviceFields();
                deviceFields.setId(IdGen.uuid());
                deviceFields.setDeviceId(deviceId);
                //省略部分業務代碼
                Gateway gateway = new Gateway();
                gateway.setId(IdGen.uuid());
                //省略部分業務代碼
                if(!deviceService.checkDeviceUidRepeat(uid)){
//                    重復，返回sheet行號，並寫入sheet表單中
                    result.add(1);
                    continue;
                }
//                關聯表一個事務處理
                boolean flg = deviceService.addDeviceEtc(device, deviceFields, gateway);
                if(!flg){
                    result.add(1);
                }else{
                    result.add(0);
                }
            }
        }
//            將反饋結果寫入文件0-成功，1-失敗
            ReadExcelUtil.writeExcelResult(url, result);
            AliYunFileSetting setting = new AliYunFileSetting();
            setting.setAccessKeyId(aliConstants.accessKeyId);
            setting.setAccessKeySecret(aliConstants.accessKeySecret);
            setting.setBucketName(aliConstants.bucketName);
            setting.setEndpoint(aliConstants.endpoint);
            AliyunFileManager manager = AliyunFileManager.getInstance(setting);
            InputStream is = new FileInputStream(tempFile);
            String relativePath =aliConstants.excelFilePath;//相對路徑
            boolean fg = manager.upload(is, relativePath, batchRecord.getFileName());//上傳文件與客戶上傳后生成的文件名相同。
            if(fg){
                logger.info("反饋已經成功寫入文件中！");
//            更新批量記錄
                batchRecord.setUpdateTime(DateUtil.getNowTimestamp());
                deviceService.updateBatchRecordByPrimaryKey(batchRecord);
            }
    } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        response.setMessage("批量導入設備成功！");
        response.setStatus(Defined.STATUS_SUCCESS);
        return response;
    }
    @PostMapping("uploadExcel")
     public ResponseObj uploadExcel(@RequestParam("excelFile") MultipartFile file,@RequestParam("companyId") String companyId,
             @RequestParam("productId") String productId,HttpServletRequest request) throws Exception {
        ResponseObj response = new ResponseObj();
        response.setData(Defined.STATUS_SUCCESS);
        response.setMessage("文件上傳成功！");
        ResponseObj resp = new ResponseObj();
        resp.setData(Defined.STATUS_ERROR);
        resp.setMessage("不是文件！");
        AliYunFileSetting setting = new AliYunFileSetting();
        setting.setAccessKeyId(aliConstants.accessKeyId);
        setting.setAccessKeySecret(aliConstants.accessKeySecret);
        setting.setBucketName(aliConstants.bucketName);
        setting.setEndpoint(aliConstants.endpoint);
        AliyunFileManager manager = AliyunFileManager.getInstance(setting);
        if(file.isEmpty()){
            response.setData(Defined.STATUS_ERROR);
            response.setMessage("不是文件！");
            return response;
           }
        String fileName = file.getOriginalFilename();
        long fileSize = file.getSize();
        File tempFile = ReadExcelUtil.createTempFile(fileName);
        String relativePath =aliConstants.excelFilePath;//相對路徑
        String dir =  aliConstants.aliyunHostOuter+"/"+relativePath;//雲端絕對路徑
        File dest = new File(dir);
        if(!dest.exists()){ //判斷文件目錄是否存在
            dest.mkdir();//
        }
        try {
            file.transferTo(tempFile); 
            InputStream is = new FileInputStream(tempFile);
            String suffix=fileName.substring(fileName.lastIndexOf("."));//獲取原始文件后綴.xlxs(含點)
            String newFileName = IdGen.uuid()+suffix;
            boolean result = manager.upload(is, relativePath, newFileName);//上傳文件，並建立隨機文件名。
//          boolean result = manager.upload(is, dir, fileName);//上傳阿里雲，固定文件名
            if(result){
                 response.setData(dir+"/"+newFileName);//返回新建文件名的絕對路徑
//                 數據庫存入相對路徑
                 BatchRecord batchRecord = new BatchRecord();
                 batchRecord.setFileName(newFileName);
                 batchRecord.setFilePath(relativePath+"/"+newFileName);
                 batchRecord.setFileSize(fileSize);
                 batchRecord.setIsDelete((byte) 0);
                 batchRecord.setStatus((byte) 0);
                 batchRecord.setId(IdGen.uuid());
                 batchRecord.setCreateTime(DateUtil.getNowTimestamp());
//                 batchRecord.setUpdateTime(DateUtil.getNowTimestamp());
                 batchRecord.setCompanyId(companyId);
                 batchRecord.setProductId(productId);
                 Integer resultNum  = deviceService.addBatchRecord(batchRecord);
                 if(resultNum>0){
                     tempFile.delete();
                 return response;
                 }
                 return resp;
            }else{
                resp.setMessage("文件上傳失敗！");
                return resp;
            }
        } catch (IllegalStateException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            resp.setMessage("文件上傳異常！");
            return resp;
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            resp.setMessage("文件上傳異常！");
            return resp;
        }
    }

模板圖片：