Apache POI操作批量导入MySQL数据库

Apache POI操作批量导入MySQL数据库

poi介绍：

Apache POI是用Java编写的免费开源的跨平台的Java API，Apache POI提供API给Java程序对Microsoft Office格式档案读和写的功能，其中使用最多的就是使用POI操作Excel文件。

POI使用到的相关maven依赖坐标如下：

<dependency><groupId>org.apache.poi</groupId><artifactId>poi</artifactId><version>3.14</version></dependency><dependency><groupId>org.apache.poi</groupId><artifactId>poi-ooxml</artifactId><version>3.14</version></dependency>

POI的相关操作结果

HSSF － 提供读写Microsoft Excel XLS格式档案的功能
XSSF － 提供读写Microsoft Excel OOXML XLSX格式档案的功能
HWPF － 提供读写Microsoft Word DOC格式档案的功能
HSLF － 提供读写Microsoft PowerPoint格式档案的功能
HDGF － 提供读Microsoft Visio格式档案的功能
HPBF － 提供读Microsoft Publisher格式档案的功能
HSMF － 提供读Microsoft Outlook格式档案的功能

1、POI操作入门案例

1.1、从Excel文件读取数据1

使用POI可以从一个已经存在的Excel文件中读取数据

前提需要建立一个需要读取的表格数据进行读取

/**
 * 使用poi读取表格数据
 * @throws Exception
 */@Testpublicvoidtest1()throwsException{// 1、加载指定的文件进行读取XSSFWorkbook excel =newXSSFWorkbook(newFileInputStream("C:\\Users\\zhong\\Desktop\\poi.xlsx"));// 2、读取表格中的Sheet页，通过索引决定XSSFSheet sheetAt = excel.getSheetAt(0);// 3、读取Sheet页中的行数据for(Row row : sheetAt){// 4、读取每一行数据的单元格数据（如果涉及到类型装转换的可能出现报错消息，后期通过代码进行判断即可）for(Cell cell : row){System.out.print(cell+"   ");}System.out.println();}// 5、关闭读取文件的流
    excel.close();}

输出结果如下：

姓名   省份   城市   
张三   广东   高州   
李四   四川   成都   

POI操作Excel表格封装了几个核心对象：

XSSFWorkbook：工作簿
XSSFSheet：工作表
Row：行
Cell：单元格

1.2、从Excel文件读取数据2

还有一种方式就是获取工作表最后一个行号，从而根据行号获得行对象，通过行获取最后一个单元格索引，从而根据单元格索引获取每行的一个单元格对象，代码如下：

/**
 * 使用poi读取文件的第二种方式
 * @throws Exception
 */@Testpublicvoidtest2()throwsException{// 1、加载指定的文件进行读取XSSFWorkbook excel =newXSSFWorkbook(newFileInputStream("C:\\Users\\zhong\\Desktop\\poi.xlsx"));// 2、读取表格中的Sheet页，通过索引决定XSSFSheet sheetAt = excel.getSheetAt(0);// 3、获取当前工作表中最后一个行号,注意行号是从0开启的int lastRowNum = sheetAt.getLastRowNum();// 4、遍历获取到的行号for(int i =0; i <= lastRowNum; i++){// 5、根据行号获取到每一行的数据XSSFRow row = sheetAt.getRow(i);// 6、获取到当前最后一个单元格索引short lastCellNum = row.getLastCellNum();// 7、遍历当前的单元格获取到具体的数据for(int j =0; j < lastCellNum; j++){// 获取到单元格的对象XSSFCell cell = row.getCell(j);// 输出获取到的数据System.out.print(cell +"   ");}System.out.println();}// 8、释放资源
    excel.close();}

1.3、向Excel文件写入数据

使用POI可以在内存中创建一个Excel文件并将数据写入到这个文件，最后通过输出流将内存中的Excel文件下载到磁盘

/**
 * poi写出数据到磁盘
 */@Testpublicvoidtest3()throwsException{// 1、创建工作簿XSSFWorkbook excel =newXSSFWorkbook();// 2、创建工作簿中的表对象XSSFSheet sheet = excel.createSheet("创建表");// 3、创建第一行（表头）XSSFRow row1 = sheet.createRow(0);// 4、在行中创建单元格数据
    row1.createCell(0).setCellValue("姓名");
    row1.createCell(1).setCellValue("省份");
    row1.createCell(2).setCellValue("城市");
    row1.createCell(3).setCellValue("年龄");// 5、创建第二行XSSFRow row2 = sheet.createRow(1);// 6、创建第6行的数据
    row2.createCell(0).setCellValue("张三");
    row2.createCell(1).setCellValue("辽宁");
    row2.createCell(2).setCellValue("上海");
    row2.createCell(3).setCellValue("50");// 7、创建一个字节输出流，将数据保存到本地FileOutputStream fileOutputStream =newFileOutputStream(newFile("C:\\Users\\zhong\\Desktop\\aaa.xlsx"));
    excel.write(fileOutputStream);
    fileOutputStream.flush();// 8、关闭输出
    excel.close();System.out.println("数据导出成功");}

2、使用POI批量导入数据到MySQL数据库

创建一个数据库的表

t_ordersetting

-- auto-generated definitioncreatetable t_ordersetting
(
    id           intauto_incrementprimarykey,
    orderDate    datenullcomment'约预日期',
    number       intnullcomment'可预约人数',
    reservations intnullcomment'已预约人数')charset= utf8;

批量导入预约设置信息操作过程：

1、点击模板下载按钮下载Excel模板文件

2、将预约设置信息录入到模板文件中

3、点击上传文件按钮将录入完信息的模板文件上传到服务器

4、通过POI读取上传文件的数据并保存到数据库

创建对应的实体类

publicclassOrderSettingimplementsSerializable{privateInteger id ;privateDate orderDate;//预约设置日期privateint number;//可预约人数privateint reservations ;//已预约人数publicOrderSetting(){}publicOrderSetting(Date orderDate,int number){this.orderDate = orderDate;this.number = number;}}

2.1、创建导入数据的Excel模板文件

目的是为了统一管理导入的数据格式

修改页面提供一个下载批量导入数据的模板按钮

//下载模板文件downloadTemplate(){
    window.location.href="../../template/ordersetting_template.xlsx";}

文件上传前端代码实现

<el-uploadaction="/ordersetting/upload.do"name="excelFile":show-file-list="false":on-success="handleSuccess":before-upload="beforeUpload"><el-buttontype="primary">上传文件</el-button></el-upload>

提交函数

// 上传之前进行文件格式校验beforeUpload(file){const isXLS = file.type ==='application/vnd.ms-excel';if(isXLS){returntrue;}const isXLSX = file.type ==='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet';if(isXLSX){returntrue;}this.$message.error('上传文件只能是xls或者xlsx格式!');returnfalse;},// 上传成功提示handleSuccess(response, file){if(response.flag){this.$message({message: response.message,type:'success'});}else{this.$message.error(response.message);}
    console.log(response, file, fileList);},

2.2、批量上传文件的后端代码编写

关于上传文件一般网上都会有很多的工具类可以下载使用，也就是别人将已经封装好的代码拿出来分享给大家使用的

packagecom.zcl.utils;importjava.io.FileNotFoundException;importjava.io.IOException;importjava.io.InputStream;importjava.text.SimpleDateFormat;importjava.util.ArrayList;importjava.util.List;importorg.apache.poi.hssf.usermodel.HSSFWorkbook;importorg.apache.poi.ss.usermodel.Cell;importorg.apache.poi.ss.usermodel.Row;importorg.apache.poi.ss.usermodel.Sheet;importorg.apache.poi.ss.usermodel.Workbook;importorg.apache.poi.xssf.usermodel.XSSFWorkbook;importorg.springframework.web.multipart.MultipartFile;publicclassPOIUtils{privatefinalstaticString xls ="xls";privatefinalstaticString xlsx ="xlsx";privatefinalstaticString DATE_FORMAT ="yyyy/MM/dd";/**
     * 读入excel文件，解析后返回
     * @param file
     * @throws IOException
     */publicstaticList<String[]>readExcel(MultipartFile file)throwsIOException{//检查文件checkFile(file);//获得Workbook工作薄对象Workbook workbook =getWorkBook(file);//创建返回对象，把每行中的值作为一个数组，所有行作为一个集合返回List<String[]> list =newArrayList<String[]>();if(workbook !=null){for(int sheetNum =0;sheetNum < workbook.getNumberOfSheets();sheetNum++){//获得当前sheet工作表Sheet sheet = workbook.getSheetAt(sheetNum);if(sheet ==null){continue;}//获得当前sheet的开始行int firstRowNum  = sheet.getFirstRowNum();//获得当前sheet的结束行int lastRowNum = sheet.getLastRowNum();//循环除了第一行的所有行for(int rowNum = firstRowNum+1;rowNum <= lastRowNum;rowNum++){//获得当前行Row row = sheet.getRow(rowNum);if(row ==null){continue;}//获得当前行的开始列int firstCellNum = row.getFirstCellNum();//获得当前行的列数int lastCellNum = row.getPhysicalNumberOfCells();String[] cells =newString[row.getPhysicalNumberOfCells()];//循环当前行for(int cellNum = firstCellNum; cellNum < lastCellNum;cellNum++){Cell cell = row.getCell(cellNum);
                        cells[cellNum]=getCellValue(cell);}
                    list.add(cells);}}
            workbook.close();}return list;}//校验文件是否合法publicstaticvoidcheckFile(MultipartFile file)throwsIOException{//判断文件是否存在if(null== file){thrownewFileNotFoundException("文件不存在！");}//获得文件名String fileName = file.getOriginalFilename();//判断文件是否是excel文件if(!fileName.endsWith(xls)&&!fileName.endsWith(xlsx)){thrownewIOException(fileName +"不是excel文件");}}publicstaticWorkbookgetWorkBook(MultipartFile file){//获得文件名String fileName = file.getOriginalFilename();//创建Workbook工作薄对象，表示整个excelWorkbook workbook =null;try{//获取excel文件的io流InputStream is = file.getInputStream();//根据文件后缀名不同(xls和xlsx)获得不同的Workbook实现类对象if(fileName.endsWith(xls)){//2003
                workbook =newHSSFWorkbook(is);}elseif(fileName.endsWith(xlsx)){//2007
                workbook =newXSSFWorkbook(is);}}catch(IOException e){
            e.printStackTrace();}return workbook;}publicstaticStringgetCellValue(Cell cell){String cellValue ="";if(cell ==null){return cellValue;}//如果当前单元格内容为日期类型，需要特殊处理String dataFormatString = cell.getCellStyle().getDataFormatString();if(dataFormatString.equals("m/d/yy")){
            cellValue =newSimpleDateFormat(DATE_FORMAT).format(cell.getDateCellValue());return cellValue;}//把数字当成String来读，避免出现1读成1.0的情况if(cell.getCellType()==Cell.CELL_TYPE_NUMERIC){
            cell.setCellType(Cell.CELL_TYPE_STRING);}//判断数据的类型switch(cell.getCellType()){caseCell.CELL_TYPE_NUMERIC://数字
                cellValue =String.valueOf(cell.getNumericCellValue());break;caseCell.CELL_TYPE_STRING://字符串
                cellValue =String.valueOf(cell.getStringCellValue());break;caseCell.CELL_TYPE_BOOLEAN://Boolean
                cellValue =String.valueOf(cell.getBooleanCellValue());break;caseCell.CELL_TYPE_FORMULA://公式
                cellValue =String.valueOf(cell.getCellFormula());break;caseCell.CELL_TYPE_BLANK://空值
                cellValue ="";break;caseCell.CELL_TYPE_ERROR://故障
                cellValue ="非法字符";break;default:
                cellValue ="未知类型";break;}return cellValue;}}

2.2.1、创建批量上传控制器

这里使用的dubbo远程调用了接口方法

/**
 * 项目名称：health_parent
 * 描述：预约管理控制器
 *
 * @author zhong
 * @date 2022-06-18 14:50
 */@RestController@RequestMapping("/ordersetting")publicclassOrderSettingController{/**
     * 远程调用预约管理的服务接口
     */@ReferenceprivateOrderSettingService orderSettingService;/**
     * 文件上传，实现预约管理的批量导入
     * @param excelFile
     * @return
     */@RequestMapping("/upload")publicResultupload(@RequestParam("excelFile")MultipartFile excelFile){try{// 1、使用工具类解析上传的数据List<String[]> list =POIUtils.readExcel(excelFile);// 将读取的表格数据转换为表格预约对象数据List<OrderSetting> data =newArrayList<>();// 2、处理解析的数据for(String[] strings : list){// 获取到导入数据行的第一个单元格(预约时间)String orderSate = strings[0];// 获取到导入数据行的第二个单元格（预约姓名）String number = strings[1];// 将重新获取到的数据添加到预约集合中OrderSetting orderSetting =newOrderSetting(newDate(orderSate),Integer.parseInt(number));
                data.add(orderSetting);}// 2、通过dubbo远程调用服务实现数据批量导入到数据库
            orderSettingService.add(data);}catch(IOException e){
            e.printStackTrace();returnnewResult(false,MessageConstant.IMPORT_ORDERSETTING_FAIL);}returnnewResult(true,MessageConstant.IMPORT_ORDERSETTING_SUCCESS);}}

2.2.2、创建批量上传的接口实现类

packagecom.zcl.service.impl;importcom.alibaba.dubbo.config.annotation.Service;importcom.itheima.pojo.OrderSetting;importcom.zcl.dao.OrderSettingDao;importcom.zcl.service.OrderSettingService;importorg.springframework.transaction.annotation.Transactional;importjava.util.List;/**
 * 项目名称：health_parent
 * 描述：预约管理接口实现类
 *
 * @author zhong
 * @date 2022-06-18 15:09
 */@Service(interfaceClass =OrderSettingService.class)@TransactionalpublicclassOrderSettingImplimplementsOrderSettingService{/**
     * 注入保存数据
     */@AutowiredprivateOrderSettingDao orderSettingDao;/**
     * 批量上传业务实现
     * @param data
     */@Overridepublicvoidadd(List<OrderSetting> data){// 判断集合数据是否为空if(data !=null&& data.size()>0){// 判断当前日期是否已经进行了预约设置for(OrderSetting datum : data){// 根据日期查询预约消息long count = orderSettingDao.findCountByOrderDate(datum.getOrderDate());if(count >0){// 已进行预约管理，执行更新操作
                    orderSettingDao.editNumberByOrderDate(datum);}else{// 添加预约设置
                    orderSettingDao.add(datum);}}}else{}}}

2.2.3、创建数据访问层接口和映射文件

packagecom.zcl.dao;importcom.itheima.pojo.OrderSetting;importjava.util.Date;/**
 * 项目名称：health_parent
 * 描述：预约管理数据访问层
 *
 * @author zhong
 * @date 2022-06-18 15:13
 */publicinterfaceOrderSettingDao{/**
     * 添加预约数据
     * @param orderSetting
     */voidadd(OrderSetting orderSetting);/**
     * 修改预约数据
     * @param orderSetting
     */voideditNumberByOrderDate(OrderSetting orderSetting);/**
     * 查询预约数据的总数
     * @param orderDate
     * @return
     */longfindCountByOrderDate(Date orderDate);}

数据访问层映射文件编写SQL语句

<?xml version="1.0" encoding="UTF-8"?><!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN""http://mybatis.org/dtd/mybatis-3-mapper.dtd"><mapper namespace="com.zcl.dao.OrderSettingDao"><!--添加预约嘻嘻嘻--><insert id="add" parameterType="com.itheima.pojo.OrderSetting">
        insert into t_ordersetting (orderDate,number,reservations)
        values (#{orderDate},#{number},#{reservations});</insert><!--根据日期修改预约信息--><update id="editNumberByOrderDate" parameterType="com.itheima.pojo.OrderSetting">
        update t_ordersetting
        set number = #{number}
        where orderDate = #{orderDate}</update><!--根据日期查询数据--><select id="findCountByOrderDate" parameterType="date" resultType="java.lang.Long">
        select count(id)
        from t_ordersetting
        where orderDate = #{orderDate}</select></mapper>

3、批量导入数据测试

在上传模板上制作数据,然后进行导入，先导入新的数据再导入修改后日期不变的数据再次上传数据