SpringBoot使用POI-TL生成WORD

2024-02-18 09:04:51

引言
Apache POI是一个流行的Java库，用于读写Microsoft Office格式的文档，例如Word、Excel和PowerPoint。POI-TL是POI的一个扩展，它提供了一种模板语言，允许您使用占位符在Word文档中动态插入数据。这使得在运行时生成定制的Word文档成为可能，例如合同、发票或报告。

整合SpringBoot
首先，我们需要在SpringBoot项目中添加对POI-TL的依赖。这可以通过在pom.xml文件中添加以下依赖项来实现：

<dependency>
    <groupId>org.apache.poi</groupId>
    <artifactId>poi-tl</artifactId>
    <version>5.2.2</version>
</dependency>

添加依赖后，就可以开始使用POI-TL了。

示例代码

import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.apache.poi.xwpf.usermodel.XWPFParagraph;
import org.apache.poi.xwpf.usermodel.XWPFRun;
import org.apache.poi.xwpf.usermodel.XWPFTable;
import org.apache.poi.xwpf.usermodel.XWPFTableCell;
import org.apache.poi.xwpf.usermodel.XWPFTableRow;
import org.springframework.stereotype.Service;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service
public class WordGeneratorService {

    public byte[] generateWordDocument(Map<String, Object> data) throws IOException {
        // 创建一个新的Word文档
        XWPFDocument document = new XWPFDocument();

        // 创建一个新的段落
        XWPFParagraph paragraph = document.createParagraph();

        // 在段落中添加文本
        XWPFRun run = paragraph.createRun();
        run.setText("Hello, ${name}!");

        // 将数据绑定到模板变量
        Map<String, String> templateVariables = new HashMap<>();
        templateVariables.put("name", "John Doe");

        // 使用模板引擎解析文档
        POIUtils.populateTemplate(document, templateVariables);

        // 创建一个新的表格
        XWPFTable table = document.createTable();

        // 在表格中添加标题行
        XWPFTableRow headerRow = table.getRow(0);
        headerRow.getCell(0).setText("Name");
        headerRow.getCell(1).setText("Age");

        // 在表格中添加数据行
        List<Map<String, String>> rows = (List<Map<String, String>>) data.get("rows");
        for (Map<String, String> row : rows) {
            XWPFTableRow row1 = table.createRow();
            row1.getCell(0).setText(row.get("name"));
            row1.getCell(1).setText(row.get("age"));
        }

        // 将文档写入字节数组输出流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        document.write(out);

        // 返回字节数组输出流的内容
        return out.toByteArray();
    }
}