使用Apache的POI实现职位的导入导出功能

Apache POI 是 Apache 软件基金会提供的 100% 开源库，支持 Excel 库的所有基本功能。在 POI 中，Workbook代表着一个 Excel 文件（工作簿），Sheet代表着 Workbook 中的一个表格，Row 代表 Sheet 中的一行，而 Cell 代表着一个单元格。
HSSFWorkbook对应的就是一个 .xls 文件，兼容 Office97-2003 版本。
XSSFWorkbook对应的是一个 .xlsx 文件，兼容 Office2007 及以上版本。
在HSSFWorkbook 中，Sheet接口的实现类为 HSSFSheet，Row接口的实现类为HSSFRow，Cell 接口的实现类为 HSSFCell。
XSSFWorkbook 中实现类的命名方式类似，在 Sheet、Row、Cell 前加 XSSF 前缀即可。

一.导入的代码实现：

后端代码部分：

1.添加依赖

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

2.写excel导入的方法

 public static List<Position> importData(MultipartFile file) {
        List<Position> positions = new ArrayList<>();
        Position position = null;
        try {
            XSSFWorkbook workbook = new XSSFWorkbook(file.getInputStream());
            // 2. 获取 workbook的 sheet数量
            int numberOfSheets = workbook.getNumberOfSheets();
            for (int i = 0; i < numberOfSheets; i++) {
                // 3. 获取 sheet
                XSSFSheet sheet = workbook.getSheetAt(i);
                int rows = sheet.getPhysicalNumberOfRows();
                for (int j = 1; j < rows; j++) {
                    XSSFRow row = sheet.getRow(j);
                    if (row != null) {
                        position = new Position();
                        int cellNum = row.getPhysicalNumberOfCells();
                        for (int k = 0; k < cellNum; k++) {
                            XSSFCell cell = row.getCell(k);
                            switch (cell.getCellType()) {
                                case STRING:
                                    if (k == 1) {
                                        position.setName(cell.getStringCellValue());
                                    }
                                    break;
                                case BOOLEAN:
                                    if(k == 3) {
                                        position.setEnabled(cell.getBooleanCellValue());
                                    }
                                    break;
                                default:
                                    if(k == 2) {
                                        position.setCreateDate(cell.getDateCellValue());
                                    }
                                    break;
                            }
                        }
                        positions.add(position);

                    }
                }

            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return positions;
    }

3.写批量添加的sql

在mapper中写批量添加的方法

Integer batchInsert(List<Position> positions);

在xml写批量添加的sql语句

<insert id="batchInsert" parameterType="com.niit.springboothr.model.Position">
    insert into position (name, createDate, enabled) values
    <foreach collection="list" item="position" separator=",">
      (#{position.name,jdbcType=VARCHAR}, #{position.createDate, jdbcType=TIMESTAMP},
      #{position.enabled, jdbcType=BIT})
    </foreach>
  </insert>

在service中写批量添加的方法

 public int addPositions(List<Position> positions){
        return positionMapper.batchInsert(positions);
    }

4.写导入的接口

@PostMapping("/import")
    @ApiOperation(value = "导入数据", notes = "导入excel数据")
    public RespBean importData(MultipartFile file) throws IOException {
        List<Position> positions = PoiUtils.importData(file);
        if(positionService.addPositions(positions) == positions.size()) {
            return RespBean.ok("导入成功");
        }
        return RespBean.ok("导入失败");
    }

前端代码部分：

1.在职位中添加导入按钮

<el-upload
        :show-file-list="false"
        :before-upload="beforeUpload"
        :on-success="onSuccess"
        :on-error="onError"
        :disabled="importBtnDisabled"
        style="display: inline-flex; margin-right: 8px"
        action="/system/basic/pos/import">
        <el-button type="success" :icon="importBtnIcon" size="small">{{importBtnText}}</el-button>
      </el-upload>

2.在职位中实现导入

在data中添加

importBtnDisabled: false,
importBtnText: '导入数据',
mportBtnIcon: 'el-icon-upload2',

在methods中添加

onSuccess(response, file, fileList) {
      this.importBtnText = '导入数据'
      this.importBtnIcon = 'el-icon-upload2'
      this.importBtnDisabled = false
      this.initPositions()
    },
    onError(err, file, fileList) {
      this.importBtnText = '导入数据'
      this.importBtnIcon = 'el-icon-upload2'
      this.importBtnDisabled = false
    },
    beforeUpload () {
      this.importBtnText = '正在导入'
      this.importBtnIcon = 'el-icon-loading'
      this.importBtnDisabled = true
    },

二.导出的代码实现：

后端代码部分：

1.添加依赖

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

2.写excel导出的方法

public static ResponseEntity<byte[]> exportData(List<Position> positions) {
        // 1. 创建 excel文件
        XSSFWorkbook workbook = new XSSFWorkbook();
        // 2. 创建样式
        // 标题行样式
        XSSFCellStyle headerStyle = workbook.createCellStyle();
        headerStyle.setFillForegroundColor(IndexedColors.YELLOW.index);
        headerStyle.setFillPattern(FillPatternType.SOLID_FOREGROUND);
        // 日期样式
        XSSFCellStyle dateCellStyle = workbook.createCellStyle();
        dateCellStyle.setDataFormat(HSSFDataFormat.getBuiltinFormat("m/d/yy h:mm"));

        // 3. 写入标题栏数据
        XSSFSheet sheet = workbook.createSheet("职位信息表");
        sheet.setColumnWidth(0, 5 * 256);
        sheet.setColumnWidth(1, 15 * 256);
        sheet.setColumnWidth(2, 20 * 256);
        sheet.setColumnWidth(3, 10 * 256);

        Row titleRow = sheet.createRow(0);
        Cell c0 = titleRow.createCell(0);
        Cell c1 = titleRow.createCell(1);
        Cell c2 = titleRow.createCell(2);
        Cell c3 = titleRow.createCell(3);

        c0.setCellStyle(headerStyle);
        c1.setCellStyle(headerStyle);
        c2.setCellStyle(headerStyle);
        c3.setCellStyle(headerStyle);

        c0.setCellValue("编号");
        c1.setCellValue("职位名称");
        c2.setCellValue("创建时间");
        c3.setCellValue("是否启用");

        // 4. 写入 postions数据
        for (int i = 0; i < positions.size(); i++) {
            Position position = positions.get(i);
            Row row = sheet.createRow(i + 1);
            row.createCell(0).setCellValue(position.getId());
            row.createCell(1).setCellValue(position.getName());

            Cell cell = row.createCell(2);
            cell.setCellStyle(dateCellStyle);
            cell.setCellValue(position.getCreateDate());

            row.createCell(3).setCellValue(position.getEnabled());
        }
        // 5. 保存文件
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        HttpHeaders headers = new HttpHeaders();
        try {
            headers.setContentDispositionFormData("attachment",
                    new String("aaa.xlsx".getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1));
            headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
            workbook.write(out);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return new ResponseEntity<>(out.toByteArray(), headers, HttpStatus.CREATED);
    }

3.写导出的接口

  @GetMapping("/export")
    @ApiOperation(value = "导出数据", notes = "将所有职位导出到excel")
    public ResponseEntity<byte[]> exportData() {
        List<Position> positions = positionService.getAllPosition();
        return PoiUtils.exportData(positions);
    }

前端代码部分：

1.在职位中添加导出按钮

 <el-button type="success" icon="el-icon-download" size="small" @click="pullData">导出数据</el-button>

2.在职位中实现导出

在methods中添加

 pullData() {
       window.open('/system/basic/pos/export', '_parent')
    },

三.运行结果：

1.导入

2.导出

POI导入导出功能

POI导入导出功能

使用Apache的POI实现职位的导入导出功能

一.导入的代码实现：

后端代码部分：

1.添加依赖

2.写excel导入的方法

3.写批量添加的sql

4.写导入的接口

前端代码部分：

1.在职位中添加导入按钮

2.在职位中实现导入

二.导出的代码实现：

后端代码部分：

1.添加依赖

2.写excel导出的方法

3.写导出的接口

前端代码部分：

1.在职位中添加导出按钮

2.在职位中实现导出

三.运行结果：