SSM框架基于poi实现excel文件的上传以及导入mysql数据库

时间 2019-11-06

标签 ssm 框架基于 poi 实现 excel 文件上传以及导入 mysql 数据库栏目 Apache 繁體版

原文原文链接

最近正在完善以前和小伙伴假期作的一个项目的一些小功能，其中一个就是上传excel文件而且将excel文件中的数据导入到指定数据库中。在这其中有遇到一些很小的细节的问题，因此决定将这个实现过程记录下来。前端

一.基本介绍

（1）先后台分离，前端使用form表单提交，直接将文件流传递给后台，后台经过poi进行解析。java

（2）在Excel中，有几个基础的概念web

一个Execl就是一个Workbook
一个Sheet就是一张表格
一个Workbook能够包含多个Sheet
一行为一个Row
每一行（Row）的每一列就是一个单元格（Cell）

（3）实现思路：在读取Excel文件的过程当中，咱们能够首先构造一个Workbook实例，而后遍历每个Sheet，对于每个sheet，咱们再遍历每个row，基于预先设定好的列名，读取每一个cell的值。将读取到的每个cell值，放进预先建立好的对象实例中去。【每个row就至关于一个实例对象】，最后咱们将这些实例对象组成一个对象列表，批量插入到数据库中。spring

二.代码实现

1.导入依赖包以及基础配置

<dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.8</version>
            <exclusions>
                <exclusion>
                    <artifactId>commons-codec</artifactId>
                    <groupId>commons-codec</groupId>
                </exclusion>
            </exclusions>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.8</version>
        </dependency>

        <dependency>
            <groupId>commons-fileupload</groupId>
            <artifactId>commons-fileupload</artifactId>
            <version>1.3.3</version>
        </dependency>
        <dependency>
            <groupId>commons-io</groupId>
            <artifactId>commons-io</artifactId>
            <version>2.5</version>
        </dependency>

在spring-mvc.xml中添加代码数据库

<bean id="multipartResolve" class="org.springframework.web.multipart.support.StandardServletMultipartResolver">
    </bean>

在web.xml中央控制器中添加支持apache

2.Controller层代码数组

后台使用MultipartFile来获取上传的文件。须要注意的是其中的@RequestParam中的参数值须要与前端页面表单中type="file"的name的属性值相同。spring-mvc

//导入excel用户信息
    @ResponseBody
    @RequestMapping(value = "/importUserInfo.do", method = RequestMethod.POST)
    public ResultModel importUserInfo(@RequestParam("file") MultipartFile file) throws Exception {
        if (!file.isEmpty()) {
            InputStream inputStream = file.getInputStream();
            return ResultModel.builder()
                    .data(userService.importUserInfo(inputStream, file.getOriginalFilename()))
                    .code(SUCCESS)
                    .build();
        } else
            return ResultModel.builder()
                    .data("File is empty!")
                    .code(SUCCESS)
                    .build();
    }

3.userService中的实现方法importUserInfomvc

@Override
    public Integer importUserInfo(InputStream filePath, String file) {
        List<User> list = new ArrayList<>();
        String[] s;
        if (!file.endsWith(".xls") && !file.endsWith(".xlsx")) {
            //文件上传格式出错
            return -2;
        }
        Workbook workbook = null;
        try {
            //获取Workbook实例
            workbook = ExcelUtil.getWorkBook(file, filePath);
        } catch (Exception e) {
            e.printStackTrace();
        }

        Sheet sheet = workbook.getSheetAt(0);
        
        //去除sheet中的一些无效行，好比值为空可是有格式的空白行
        sheet = ExcelUtil.resetSheet(sheet);
        int lastRowNum = sheet.getLastRowNum();
        for (int i = 1; i <= lastRowNum; i++) {
            Row row = sheet.getRow(i);
            int cells = sheet.getRow(i).getPhysicalNumberOfCells();
            
            //获取每一行的数值，并将其填入一个字符串的数组
            s = ExcelUtil.getExcelRows(row, cells);
            User user = new User();
            for (int j = 0; j < s.length; j++) {
                if (j == 0 && !s[j].equals(""))
                    user.setInstitute(s[j]);
                else if (j == 1 && !s[j].equals(""))
                    user.setMajor(s[j]);
                else if (j == 2 && !s[j].equals(""))
                    user.setClassId(Integer.getInteger(s[j]));
                else if (j == 3 && !s[j].equals(""))
                    user.setUsername(s[j]);
                else if (j == 4 && !s[j].equals(""))
                    user.setRealName(s[j]);
                else if (j == 5 && !s[j].equals(""))
                    user.setBirthday(DateUtil.parseYYYYMMDDDate(s[j]));
                else if (j == 6 && !s[j].equals(""))
                    user.setQq(s[j]);
                else if (j == 7 && !s[j].equals(""))
                    user.setTieba(s[j]);
                else if (j == 8 && !s[j].equals(""))
                    user.setWeibo(s[j]);
                else if (j == 9 && !s[j].equals(""))
                    user.setBlog(s[j]);
                user.setPassword("huhugty7tgf6f");
            }
            list.add(user);
        }
        return userDao.insertList(list);
    }

4.Excel读取类ExcelUtilapp

import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellReference;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.InputStream;

public class ExcelUtil {

    //判断文件类型是否知足要求
    public static Workbook getWorkBook(String file, InputStream fis)throws Exception{
        Workbook workbook = null;
        if(!file.endsWith(".xls") && !file.endsWith(".xlsx")){
            throw new Exception("上传表格的格式错误！");
        }
        if(file.endsWith(".xls")){
            workbook = new HSSFWorkbook(fis,true);
        }
        if(file.endsWith(".xlsx")){
            workbook = new XSSFWorkbook(fis);
        }
        return workbook;
    }

    //获取每一行的数据，而且返回一个字符串数组
    public static String[] getExcelRows(Row row , int columuNum){
        String[] s ;
        StringBuilder builder = new StringBuilder();
        for(int x = 0;x<columuNum;x++){
            Cell cell = row.getCell(x);
            builder.append(getStringCellValue(cell)+",");
        }
        s = builder.toString().split(",");
        return s;
    }

    //获取单元内的有效数据
    public static String getStringCellValue(Cell cell){
        StringBuilder sb = new StringBuilder();
        switch (cell.getCellType()){
            //数字
            case Cell.CELL_TYPE_NUMERIC:
                if(DateUtil.isCellDateFormatted(cell)){
                    sb.append(cell.getDateCellValue());
                }else {
                    //将该数字强制转化为字符串类型获取
                    cell.setCellType(Cell.CELL_TYPE_STRING);
                    sb.append(cell.getStringCellValue());
                }
                break;

            //字符串
            case Cell.CELL_TYPE_STRING:
                sb.append(cell.getStringCellValue());
                break;

            //布尔
            case Cell.CELL_TYPE_BOOLEAN:
                sb.append(cell.getBooleanCellValue());
                break;

            //公式
            case Cell.CELL_TYPE_FORMULA:
                sb.append(cell.getCellFormula());
                break;

            //空值
            case Cell.CELL_TYPE_BLANK:
                sb.append("");
                break;

            //故障
            case Cell.CELL_TYPE_ERROR:
                sb.append("");
                break;

            default:
                sb.append("");
                break;
        }
        return sb.toString();
    }

    //过滤掉表中的无心义空白行，由于getLastRowNum（）在获取Row行数时，对于表中没有值但却有格式的无心义空白行也将计入
    public static Sheet resetSheet(Sheet sheet){
        CellReference cellReference = new CellReference("A4");
        boolean flag;
        for(int i=cellReference.getRow();i<=sheet.getLastRowNum();){
            Row r = sheet.getRow(i);
            if(r == null){
                sheet.shiftRows(i+1,sheet.getLastRowNum(),-1);
                continue;
            }
            flag = false;
            for(Cell c : r){
                if(c.getCellType()!=Cell.CELL_TYPE_BLANK){
                    flag = true;
                    break;
                }
            }
            if(flag){
                i++;
                continue;
            }
            else{   //若是是空白行，没有数据，可是有格式
                if(i == sheet.getLastRowNum())
                    sheet.removeRow(r);
                else
                    sheet.shiftRows(i+1,sheet.getLastRowNum(),-1);
            }
        }
        return sheet;
    }
}

三.测试

这里我使用postman来测试后台接口：

测试结果：