通过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，成功运行！