通過POI實現上傳EXCEL的批量讀取數據寫入數據庫

　　最近公司新增功能要求導入excel，並讀取其中數據批量寫入數據庫。於是就開始了這個事情，之前的文章，記錄了上傳文件，本篇記錄如何通過POI讀取excel數據並封裝為對象上傳。

　　上代碼：

　　1、首先這是一個依賴第三方的事情，故而需要導入jar包

        <!-- read excel start -->
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.17</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.17</version>
        </dependency>
        <!-- read excel start -->            

切記下面的那個包不可忘記，否則報錯RETURN_NULL_AND_BLANK

之前我們已經上傳含有數據的excel模板，返回的是一個url，本博主對應的url的excel表內容如下：

那么我們傳入導入的excel的url來進行讀取。不做過多解釋，該有的解釋代碼注釋里面均有

package com.topband.sweepmachine.utils;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
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 com.topband.cloud.common.utils.DateFormatUtil;
import com.topband.cloud.common.utils.StringUtil;

public class ReadExcelUtil {
    
    private Logger logger = LoggerFactory.getLogger(this.getClass());
    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{
        /*
         * workbook:工作簿,就是整個Excel文檔
         * sheet:工作表
         * row:行
         * cell:單元格
         */

//        BufferedWriter bw = new BufferedWriter(new FileWriter(new File(url)));
//        支持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();
//      創建二維數組保存所有讀取到的行列數據，外層存行數據，內層存單元格數據
        List<List<String>> dataList = new ArrayList<List<String>>();
//        FormulaEvaluator formulaEvaluator = null;
//        遍歷工作簿中的sheet,第一層循環所有sheet表
        for(int index = 0;index<sheetNum;index++){
            Sheet sheet = workbook.getSheetAt(index);
            if(sheet==null){
                continue;
            }
            System.out.println("表單行數："+sheet.getLastRowNum());
//            如果當前行沒有數據跳出循環，第二層循環單sheet表中所有行
            for(int rowIndex=0;rowIndex<=sheet.getLastRowNum();rowIndex++){
                Row row = sheet.getRow(rowIndex);
//                根據文件頭可以控制從哪一行讀取，在下面if中進行控制
                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("遍歷行中cell數據:"+getCellValue(cell));
                    cellList.add(getCellValue(cell));
                    System.out.println("第"+cellIndex+"個:     cell個數："+cellList.size());
                }
                dataList.add(cellList);
                System.out.println("第"+rowIndex+"行:     共幾行："+dataList.size());
            }
            
        }
        is.close();
        return dataList;
        }
    /**
     *獲取單元格的數據,暫時不支持公式
     * 
     *
     */
        public static String getCellValue(Cell cell){
            CellType cellType = cell.getCellTypeEnum();
            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)))){
            System.out.println(file.isFile()+"==="+file.getName().endsWith(EXCEL_XLS)+"==="+file.getName().endsWith(EXCEL_XLSX));
            System.out.println(file.getName());
            message = "文件不是Excel";
            throw new Exception(message);
        }
    }
/*    public static void main(String[] args) throws Exception {
        readExcelInfo("g://批量新增設備表.xlsx");
    }*/
}

封裝為對象，插入數據庫，這里本博主不放入公司業務對象，刪掉了

    @PostMapping("/addBatchDevice")
    public ResponseObj addBatchDevice(@RequestBody JSONObject jsonObject){
        ResponseObj response = new ResponseObj();
        response.setData(Defined.STATUS_SUCCESS);
        response.setMessage("插入成功！");
        String url = jsonObject.getString("url");
//        存放封裝的設備
        List<Device> devices = new ArrayList<Device>();
        try {
            List<List<String>> list = ReadExcelUtil.readExcelInfo(url);
            for(int i=0;i<list.size();i++){
//                new一個對象按照相應的字段設置進去就可以了，這里省略對象設置值，字段如下：
                System.out.println("封裝成對象后："+"\t"+"設備名稱--->"+list.get(i).get(0)+"型號--->"+list.get(i).get(1)+"數量---> "+list.get(i).get(2));
            }
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
//        插入數據庫
//        調用相關插入方法，可以批量也可單條插入循環實現，看具體業務需要選擇
        return response;
    }

調用接口運行效果如下：

 OK，成功運行！