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