devbox
盐析白兔
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

使用POI 导出word模板文件_poi根据模板导出word

作者：盐析白兔 | 2024-03-25 07:23:16

赞

踩

poi根据模板导出word

maven依赖

  1. <dependency>
  2.      <groupId>org.apache.poi</groupId>
  3.      <artifactId>poi</artifactId>
  4.      <version>3.16</version>
  5. </dependency>

自定义XWPFDocument 类，重写方法插入图片：

  1. /**
  2.  * @BelongsProject: exchange
  3.  * @BelongsPackage: com.elens.util
  4.  * @Author: xuweichao
  5.  * @CreateTime: 2019-03-20 12:34
  6.  * @Description: 重写XWPFDocument的方法，插入图片
  7.  */
  8. public class CustomXWPFDocument extends XWPFDocument {
  9.     public CustomXWPFDocument() {
  10.         super();
  11.     }
  12.  
  13.     public CustomXWPFDocument(OPCPackage opcPackage) throws IOException {
  14.         super(opcPackage);
  15.     }
  16.  
  17.     public CustomXWPFDocument(InputStream in) throws IOException {
  18.         super(in);
  19.     }
  20.  
  21.     public void createPicture(XWPFRun run,String blipId, int id, int width, int height) {
  22.         final int EMU = 9525;
  23.         width *= EMU;
  24.         height *= EMU;
  25.           //旧版本方法 .getPackageRelationship() 在该依赖包下已被删除
  26.         //String blipId = getAllPictures().get(id).getPackageRelationship().getId();
  27.  
  28.           //在docment下创建XWPFRun 图片会被添加到文档末尾
  29. //        CTInline inline = createParagraph().createRun().getCTR().addNewDrawing().addNewInline();
  30.         CTInline inline =run.getCTR().addNewDrawing().addNewInline();
  31.  
  32.         String picXml = "" +
  33.                 "<a:graphic xmlns:a=\"http://schemas.openxmlformats.org/drawingml/2006/main\">" +
  34.                 "   <a:graphicData uri=\"http://schemas.openxmlformats.org/drawingml/2006/picture\">" +
  35.                 "      <pic:pic xmlns:pic=\"http://schemas.openxmlformats.org/drawingml/2006/picture\">" +
  36.                 "         <pic:nvPicPr>" +
  37.                 "            <pic:cNvPr id=\"" + id + "\" name=\"Generated\"/>" +
  38.                 "            <pic:cNvPicPr/>" +
  39.                 "         </pic:nvPicPr>" +
  40.                 "         <pic:blipFill>" +
  41.                 "            <a:blip r:embed=\"" + blipId + "\" xmlns:r=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships\"/>" +
  42.                 "            <a:stretch>" +
  43.                 "               <a:fillRect/>" +
  44.                 "            </a:stretch>" +
  45.                 "         </pic:blipFill>" +
  46.                 "         <pic:spPr>" +
  47.                 "            <a:xfrm>" +
  48.                 "               <a:off x=\"0\" y=\"0\"/>" +
  49.                 "               <a:ext cx=\"" + width + "\" cy=\"" + height + "\"/>" +
  50.                 "            </a:xfrm>" +
  51.                 "            <a:prstGeom prst=\"rect\">" +
  52.                 "               <a:avLst/>" +
  53.                 "            </a:prstGeom>" +
  54.                 "         </pic:spPr>" +
  55.                 "      </pic:pic>" +
  56.                 "   </a:graphicData>" +
  57.                 "</a:graphic>";
  58.  
  59.         //CTGraphicalObjectData graphicData = inline.addNewGraphic().addNewGraphicData();
  60.         XmlToken xmlToken = null;
  61.         try {
  62.             xmlToken = XmlToken.Factory.parse(picXml);
  63.         } catch (XmlException xe) {
  64.             xe.printStackTrace();
  65.         }
  66.         inline.set(xmlToken);
  67.         //graphicData.set(xmlToken);
  68.  
  69.         inline.setDistT(0);
  70.         inline.setDistB(0);
  71.         inline.setDistL(0);
  72.         inline.setDistR(0);
  73.  
  74.         CTPositiveSize2D extent = inline.addNewExtent();
  75.         extent.setCx(width);
  76.         extent.setCy(height);
  77.  
  78.         CTNonVisualDrawingProps docPr = inline.addNewDocPr();
  79.         docPr.setId(id);
  80.         docPr.setName("Picture " + id);
  81.         docPr.setDescr("Generated");
  82.     }
  83. }

文件导出工具类：

  1. public class POIWordUtil {
  2.  
  3.     /**
  4.      * 根据模板生成新word文档
  5.      * 判断表格是需要替换还是需要插入，判断逻辑有$为替换，表格无$为插入
  6.      *
  7.      * @param inputUrl  模板存放地址
  8.      * @param outputUrl 新文档存放地址
  9.      * @param textMap   需要替换的信息集合
  10.      * @param tableList 需要插入的表格信息集合
  11.      * @return 成功返回true, 失败返回false
  12.      */
  13.     public static boolean changWord(String inputUrl, String outputUrl,
  14.                                     Map<String, Object> textMap, List<String[]> tableList) {
  15.  
  16.         //模板转换默认成功
  17.         boolean changeFlag = true;
  18.         try {
  19.             //获取docx解析对象
  20.             CustomXWPFDocument document = new CustomXWPFDocument(POIXMLDocument.openPackage(inputUrl));
  21.             //解析替换文本段落对象
  22.             POIWordUtil.changeText(document, textMap);
  23.             //解析替换表格对象
  24.             POIWordUtil.changeTable(document, textMap, tableList);
  25.  
  26.             //生成新的word
  27.             File file = new File(outputUrl);
  28.             if (!file.getParentFile().exists()) {
  29.                 file.getParentFile().mkdirs();
  30.             }
  31.             FileOutputStream stream = new FileOutputStream(file);
  32.             document.write(stream);
  33.             stream.close();
  34.  
  35.         } catch (IOException e) {
  36.             e.printStackTrace();
  37.             changeFlag = false;
  38.         }
  39.  
  40.         return changeFlag;
  41.  
  42.     }
  43.  
  44.     /**
  45.      * 替换段落文本
  46.      *
  47.      * @param document docx解析对象
  48.      * @param textMap  需要替换的信息集合
  49.      */
  50.     public static void changeText(XWPFDocument document, Map<String, Object> textMap) {
  51.         //获取段落集合
  52.         List<XWPFParagraph> paragraphs = document.getParagraphs();
  53.  
  54.         for (XWPFParagraph paragraph : paragraphs) {
  55.             //判断此段落时候需要进行替换
  56.             String text = paragraph.getText();
  57.             if (checkText(text)) {
  58.                 List<XWPFRun> runs = paragraph.getRuns();
  59.                 for (XWPFRun run : runs) {
  60.                     //替换模板原来位置
  61.                     run.setText((String) changeValue(run.toString(), textMap), 0);
  62.                 }
  63.             }
  64.         }
  65.  
  66.     }
  67.  
  68.     /**
  69.      * 替换表格对象方法
  70.      *
  71.      * @param document  docx解析对象
  72.      * @param textMap   需要替换的信息集合
  73.      * @param tableList 需要插入的表格信息集合
  74.      */
  75.     public static void changeTable(CustomXWPFDocument document, Map<String, Object> textMap,
  76.                                    List<String[]> tableList) {
  77.         //获取表格对象集合
  78.         List<XWPFTable> tables = document.getTables();
  79.         for (int i = 0; i < tables.size(); i++) {
  80.             //只处理行数大于等于2的表格，且不循环表头
  81.             XWPFTable table = tables.get(i);
  82.             if (table.getRows().size() > 1) {
  83.                 //判断表格是需要替换还是需要插入，判断逻辑有$为替换，表格无$为插入
  84.                 if (checkText(table.getText())) {
  85.                     List<XWPFTableRow> rows = table.getRows();
  86.                     //遍历表格,并替换模板
  87.                     eachTable(document, rows, textMap);
  88.                 } else {
  89. //                  System.out.println("插入"+table.getText());
  90.                     insertTable(table, tableList);
  91.                 }
  92.             }
  93.         }
  94.     }
  95.  
  96.  
  97.     /**
  98.      * 遍历表格，包含对图片内容的替换
  99.      *
  100.      * @param rows    表格行对象
  101.      * @param textMap 需要替换的信息集合
  102.      */
  103.     public static void eachTable(CustomXWPFDocument document, List<XWPFTableRow> rows, Map<String, Object> textMap) {
  104.         for (XWPFTableRow row : rows) {
  105.             List<XWPFTableCell> cells = row.getTableCells();
  106.             for (XWPFTableCell cell : cells) {
  107.                 //判断单元格是否需要替换
  108.                 if (checkText(cell.getText())) {
  109.                     List<XWPFParagraph> paragraphs = cell.getParagraphs();
  110.                     for (XWPFParagraph paragraph : paragraphs) {
  111.                         List<XWPFRun> runs = paragraph.getRuns();
  112.                         for (XWPFRun run : runs) {
  113. //                            run.setText(changeValue(run.toString(), textMap),0);
  114.  
  115.                             Object ob = changeValue(run.toString(), textMap);
  116.                             if (ob instanceof String) {
  117.                                 run.setText((String) ob, 0);
  118.                             } else if (ob instanceof Map) {
  119.                                 run.setText("", 0);
  120.  
  121.                                 Map pic = (Map) ob;
  122.                                 int width = Integer.parseInt(pic.get("width").toString());
  123.                                 int height = Integer.parseInt(pic.get("height").toString());
  124.                                 int picType = getPictureType(pic.get("type").toString());
  125.                                 byte[] byteArray = (byte[]) pic.get("content");
  126.                                 ByteArrayInputStream byteInputStream = new ByteArrayInputStream(byteArray);
  127.                                 try {
  128.  
  129.                                     String ind = document.addPictureData(byteInputStream, picType);
  130.                                     document.createPicture(run, ind, document.getNextPicNameNumber(picType), width, height);
  131.  
  132.                                 } catch (Exception e) {
  133.                                     e.printStackTrace();
  134.                                 }
  135.                             }
  136.                         }
  137.                     }
  138.                 }
  139.             }
  140.         }
  141.     }
  142.  
  143.     /**
  144.      * 为表格插入数据，行数不够添加新行
  145.      *
  146.      * @param table     需要插入数据的表格
  147.      * @param tableList 插入数据集合
  148.      */
  149.     public static void insertTable(XWPFTable table, List<String[]> tableList) {
  150.         //创建行,根据需要插入的数据添加新行，不处理表头
  151.         for (int i = 1; i < tableList.size(); i++) {
  152.             XWPFTableRow row = table.createRow();
  153.         }
  154.         //遍历表格插入数据
  155.         List<XWPFTableRow> rows = table.getRows();
  156.         for (int i = 1; i < rows.size(); i++) {
  157.             XWPFTableRow newRow = table.getRow(i);
  158.             List<XWPFTableCell> cells = newRow.getTableCells();
  159.             for (int j = 0; j < cells.size(); j++) {
  160.                 XWPFTableCell cell = cells.get(j);
  161.                 cell.setText(tableList.get(i - 1)[j]);
  162.             }
  163.         }
  164.     }
  165.  
  166.    /**
  167.      * word跨列合并单元格
  168.      */
  169.     public static void mergeCellsHorizontal(XWPFTable table, int row, int fromCell, int toCell) {
  170.         for (int cellIndex = fromCell; cellIndex <= toCell; cellIndex++) {
  171.             XWPFTableCell cell = table.getRow(row).getCell(cellIndex);
  172.             if (cellIndex == fromCell) {
  173.                 // The first merged cell is set with RESTART merge value
  174.                 cell.getCTTc().addNewTcPr().addNewHMerge().setVal(STMerge.RESTART);
  175.             } else {
  176.                 // Cells which join (merge) the first one, are set with CONTINUE
  177.                 cell.getCTTc().addNewTcPr().addNewHMerge().setVal(STMerge.CONTINUE);
  178.             }
  179.         }
  180.     }
  181.  
  182.     /**
  183.      * word跨行并单元格
  184.      */
  185.     public static void mergeCellsVertically(XWPFTable table, int col, int fromRow, int toRow) {
  186.         for (int rowIndex = fromRow; rowIndex <= toRow; rowIndex++) {
  187.             XWPFTableCell cell = table.getRow(rowIndex).getCell(col);
  188.             if (rowIndex == fromRow) {
  189.                 // The first merged cell is set with RESTART merge value
  190.                 cell.getCTTc().addNewTcPr().addNewVMerge().setVal(STMerge.RESTART);
  191.             } else {
  192.                 // Cells which join (merge) the first one, are set with CONTINUE
  193.                 cell.getCTTc().addNewTcPr().addNewVMerge().setVal(STMerge.CONTINUE);
  194.             }
  195.         }
  196.     }
  197.  
  198.     /**
  199.      * 单元格设置字体
  200.      * @param cell
  201.      * @param cellText
  202.      */
  203.     private static void getParagraph(XWPFTableCell cell,String cellText){
  204.         CTP ctp = CTP.Factory.newInstance();
  205.         XWPFParagraph p = new XWPFParagraph(ctp, cell);
  206.         p.setAlignment(ParagraphAlignment.CENTER);
  207.         XWPFRun run = p.createRun();
  208.         run.setText(cellText);
  209.         CTRPr rpr = run.getCTR().isSetRPr() ? run.getCTR().getRPr() : run.getCTR().addNewRPr();
  210.         CTFonts fonts = rpr.isSetRFonts() ? rpr.getRFonts() : rpr.addNewRFonts();
  211.         fonts.setAscii("仿宋");
  212.         fonts.setEastAsia("仿宋");
  213.         fonts.setHAnsi("仿宋");
  214.         cell.setParagraph(p);
  215.     }
  216.  
  217.  
  218.     /**
  219.      * 判断文本中时候包含$
  220.      *
  221.      * @param text 文本
  222.      * @return 包含返回true, 不包含返回false
  223.      */
  224.     public static boolean checkText(String text) {
  225.         boolean check = false;
  226.         if (text.indexOf("$") != -1) {
  227.             check = true;
  228.         }
  229.         return check;
  230.     }
  231.  
  232.     /**
  233.      * 匹配传入信息集合与模板
  234.      *
  235.      * @param value   模板需要替换的区域
  236.      * @param textMap 传入信息集合
  237.      * @return 模板需要替换区域信息集合对应值
  238.      */
  239.     public static Object changeValue(String value, Map<String, Object> textMap) {
  240.         Object obj = null;
  241.         Set<Map.Entry<String, Object>> textSets = textMap.entrySet();
  242.         for (Map.Entry<String, Object> textSet : textSets) {
  243.             //匹配模板与替换值 格式${key}
  244.             String key = "${" + textSet.getKey() + "}";
  245.             if (value.indexOf(key) != -1) {
  246.                 obj = textSet.getValue();
  247.             }
  248.         }
  249.  
  250.         if (!(obj instanceof Map)) {
  251.             //模板未匹配到区域替换为空
  252.             if (obj != null && checkText(obj.toString())) {
  253.                 obj = "";
  254.             }
  255.         }
  256.         return obj;
  257.     }
  258.  
  259.     /**
  260.      * 根据图片类型，取得对应的图片类型代码
  261.      *
  262.      * @param picType
  263.      * @return int
  264.      */
  265.     private static int getPictureType(String picType) {
  266.         int res = CustomXWPFDocument.PICTURE_TYPE_PICT;
  267.         if (picType != null) {
  268.             if (picType.equalsIgnoreCase("png")) {
  269.                 res = CustomXWPFDocument.PICTURE_TYPE_PNG;
  270.             } else if (picType.equalsIgnoreCase("dib")) {
  271.                 res = CustomXWPFDocument.PICTURE_TYPE_DIB;
  272.             } else if (picType.equalsIgnoreCase("emf")) {
  273.                 res = CustomXWPFDocument.PICTURE_TYPE_EMF;
  274.             } else if (picType.equalsIgnoreCase("jpg") || picType.equalsIgnoreCase("jpeg")) {
  275.                 res = CustomXWPFDocument.PICTURE_TYPE_JPEG;
  276.             } else if (picType.equalsIgnoreCase("wmf")) {
  277.                 res = CustomXWPFDocument.PICTURE_TYPE_WMF;
  278.             }
  279.         }
  280.         return res;
  281.     }
  282.  
  283.     /**
  284.      * 将输入流中的数据写入字节数组
  285.      *
  286.      * @param in
  287.      * @return
  288.      */
  289.     public static byte[] inputStream2ByteArray(InputStream in, boolean isClose) {
  290.         byte[] byteArray = null;
  291.         try {
  292.             int total = in.available();
  293.             byteArray = new byte[total];
  294.             in.read(byteArray);
  295.         } catch (IOException e) {
  296.             e.printStackTrace();
  297.         } finally {
  298.             if (isClose) {
  299.                 try {
  300.                     in.close();
  301.                 } catch (Exception e2) {
  302.                     System.out.println("关闭流失败");
  303.                 }
  304.             }
  305.         }
  306.         return byteArray;
  307.     }
  308.  
  309. }

功能测试：

  1.         String file =  "H:/export_template.docx";//doc 模板文件位置
  2.  
  3.         String outputUrl =  "H:/doc/test.doc";//文件输出地址
  4.  
  5.         Map<String, Object> testMap = new HashMap<>();
  6.      
  7.         testMap.put("country", data.get("country").toString());
  8.         testMap.put("birthday", data.get("birthday").toString());
  9.         testMap.put("en_name", data.get("en_name").toString());
  10.         testMap.put("enName", data.get("en_name").toString());
  11.         testMap.put("gender", data.get("gender").toString());
  12.         testMap.put("cn_name", data.get("cn_name").toString());
  13.         testMap.put("introduce", data.get("introduce").toString());
  14.         testMap.put("tags", data.get("tags").toString());
  15.         testMap.put("main_related_china", data.get("main_related_china").toString());
  16.         testMap.put("recent_related_china", data.get("recent_related_china").toString());
  17.         testMap.put("type1", data.get("type1").toString());
  18.         testMap.put("type2", data.get("type2").toString());
  19.         testMap.put("recommend_org", data.get("recommend_org").toString());
  20.         testMap.put("url_baike", data.get("url_baike").toString());
  21.         testMap.put("url_facebook", data.get("url_facebook").toString());
  22.         testMap.put("url_twitter", data.get("url_twitter").toString());
  23.         testMap.put("url_wiki", data.get("url_wiki").toString());
  24.         testMap.put("url_linkedin", data.get("url_linkedin").toString());
  25.  
  26.         testMap.put("url_blog", (String) data.get("url_blog");
  27.         String photo = data.get("photo").toString();
  28.        
  29.         Map<String, Object> header = new HashMap<>();
  30.         header.put("width", 150);
  31.         header.put("height", 200);
  32.         header.put("type", photo.substring(photo.indexOf(".") + 1));
  33.            
  34.         FileInputStream in = new FileInputStream("H:/img/test.png");
  35.  
  36.         header.put("content", POIWordUtil.inputStream2ByteArray(in, true));
  37.         testMap.put("photo", header);
  38.            
  39.         POIWordUtil.changWord(file, outputUrl, testMap, null);

我这里数据data 是从es 取得,测试的时候搞点假数据就好，路径修改成自己的。

这里${en_name} 是提前占位，具体逻辑实现可以看一下工具类。

模板如下图：

最终效果是这样的

这里参考了网上的一些资料，但是没有达到预期的效果，通过对代码的解读做出修改，已在项目中应用，效果良好。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/307921
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号