打印模板 打印流程审批意见

许鹏程

2023-08-04 a26fbe729817216bcf53b13945fe6300caa89997

 src/main/java/com/product/print/util/DynamicTableRenderPolicy.java

@@ -7,8 +7,13 @@
import com.product.common.lang.StringUtils;
import com.product.core.exception.BaseException;
import com.product.print.config.CmnCode;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.xwpf.usermodel.*;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.STVerticalAlignRun;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
@@ -58,53 +63,113 @@
      List<Map<String, Object>> subTableData = (List<Map<String, Object>>) data;
      List<XWPFTableRow> rows = xwpfTable.getRows();
      //读取rows中的内容
      //获取最后一行
      XWPFTableRow xwpfTableRow = rows.get(rows.size() - 1);
      String tableExpression = "{{" + this.replaceKey + "}}";
      //获取表格起始行和结束行
      int startRowIndex = -1;
      int endRowIndex = -1;
      rows:
      for (int i = 0; i < rows.size(); i++) {
         XWPFTableRow row = rows.get(i);
         for (int j = 0; j < row.getTableCells().size(); j++) {
            if (startRowIndex > -1 && endRowIndex > -1) {
               break rows;
            }
            XWPFTableCell cell = row.getTableCells().get(j);
            String text = cell.getText();
            if (tableExpression.equals(text)) {
               startRowIndex = i;
            }
            if ("{{$~end~}}".equals(text)) {
               endRowIndex = i;
               break rows;
            }

         }
      }
      //表格起始行不在第一行且没有找到结束标记
      if (startRowIndex > 0 && endRowIndex == -1) {
         throw new BaseException(CmnCode.NOT_FIND_END_FLAG);
      if (startRowIndex == -1) {
         throw new BaseException(CmnCode.NOT_FIND_START_FLAG);
      }

      if (startRowIndex == 0 && endRowIndex > 0) {
         //删除结束标记所在行
         xwpfTable.removeRow(endRowIndex);
      //在表格中查找子表字段以{{$开头}}以}}结尾的内容
      String regex = "\\{\\{\\$[a-zA-Z0-9_]+\\}\\}";
      Pattern pattern = Pattern.compile(regex);
      //字段所在行
      XWPFTableRow fieldRow = null;
      for (int i = startRowIndex; i < rows.size(); i++) {
         XWPFTableRow row = rows.get(i);
         for (int j = 0; j < row.getTableCells().size(); j++) {
            XWPFTableCell cell = row.getTableCells().get(j);
            String text = cell.getText();
            Matcher matcher = pattern.matcher(text);
            if (matcher.find()) {
               fieldRow = row;
               i = rows.size();
               break;
            }
         }
      }
      if (fieldRow == null) {
         throw new BaseException(CmnCode.NOT_FIND_CHILD_TABLE_FIELD);
      }

      //读取最后行每个单元格的值，调用getReplaceKey方法获取表达式中的值
      String[] fieldNames = new String[xwpfTableRow.getTableCells().size()];
//      for (int i = 0; i < xwpfTableRow.getTableCells().size(); i++) {
//         XWPFTableCell cell = xwpfTableRow.getTableCells().get(i);
//         String text = cell.getText();
//         String replaceKey = getReplaceKey(text);
//         //获取表达式中的值
//         if (StringUtils.isEmpty(replaceKey)) {
//            //设置单元格为空值
//            replaceKey = "";
//         }
//         fieldNames[i] = replaceKey;
//      }
      String[] fieldNames = new String[fieldRow.getTableCells().size()];
      for (int i = 0; i < fieldRow.getTableCells().size(); i++) {
         XWPFTableCell cell = fieldRow.getTableCells().get(i);
         String text = cell.getText();
         String replaceKey = getReplaceKey(text);
         //获取表达式中的值
         if (StringUtils.isEmpty(replaceKey)) {
            //设置单元格为空值
            replaceKey = "";
         }
         fieldNames[i] = replaceKey;
      }
      //获取fieldRow所在的下标
      int fieldRowIndex = xwpfTable.getRows().indexOf(fieldRow);

      for (int i = 0; i < subTableData.size(); i++) {
         Map<String, Object> map = subTableData.get(i);
         //创建一行在fieldRowIndex下面
         XWPFTableRow row = xwpfTable.insertNewTableRow(fieldRowIndex + 1 + i);
         copyTableRow(row, fieldRow);
         //设置row的属性与fieldRow一致
         row.setHeight(fieldRow.getHeight());
         //遍历字段每个字段创建一个单元格
         for (int j = 0; j < fieldNames.length; j++) {
            //当前单元格
            XWPFTableCell cell;
            //判断row中第j个单元格是否存在
            if (row.getTableCells().size() > j) {
               cell = row.getCell(j);
            } else {
               cell = row.createCell();
            }
            //设置单元格的值从map中取出
            //判断是否是序号列
            if (indexKey.equals(fieldNames[j])) {
               cell.setText(String.valueOf(i + 1));
               continue;
            }
            Object value = map.get(fieldNames[j]);
            if (value == null) {
               value = "";
            }
            String text = cell.getText();
            //删除单元格中的旧内容
            if (!StringUtils.isEmpty(text)) {
               List<XWPFParagraph> paragraphs = cell.getParagraphs();
               if(paragraphs.size()>1){
                  cell.removeParagraph(1);
               }
               List<XWPFRun> runs = paragraphs.get(0).getRuns();
               //清空文字
               for (int k = 0; k < runs.size(); k++) {
                  runs.get(k).setText("", 0);
               }
            }
            cell.setText(value.toString());

         }
      }
      //删除起始行
      xwpfTable.removeRow(startRowIndex);
      //删除fieldRow
      xwpfTable.removeRow(fieldRowIndex - 1);

//      //读取完毕后删除最后一行
//      xwpfTable.removeRow(rows.size() - 1);
      //获取所有的行判断单元格是否有值没有就删除该行
@@ -174,4 +239,125 @@
      }
      return null;
   }

   /**
    * 复制行，从source到target
    *
    * @param target
    * @param source
    */
   public void copyTableRow(XWPFTableRow target, XWPFTableRow source) {
      // 复制样式
      if (source.getCtRow() != null) {
         target.getCtRow().setTrPr(source.getCtRow().getTrPr());
      }
      // 复制单元格
      for (int i = 0; i < source.getTableCells().size(); i++) {
         XWPFTableCell cell1 = target.getCell(i);
         XWPFTableCell cell2 = source.getCell(i);
         if (cell1 == null) {
            cell1 = target.addNewTableCell();
         }
         copyTableCell(cell1, cell2);
      }

   }

   /**
    * 复制单元格，从source到target
    *
    * @param target
    * @param source
    */
   public void copyTableCell(XWPFTableCell target, XWPFTableCell source) {
      // 列属性
      if (source.getCTTc() != null) {
         target.getCTTc().setTcPr(source.getCTTc().getTcPr());
      }
      // 删除段落
      for (int pos = 0; pos < target.getParagraphs().size(); pos++) {
         target.removeParagraph(pos);
      }
      // 添加段落
      for (XWPFParagraph sp : source.getParagraphs()) {
         XWPFParagraph targetP = target.addParagraph();
         copyParagraph(targetP, sp);
      }
   }

   /**
    * 复制图片到target
    *
    * @param target
    * @param picture
    * @throws IOException
    * @throws InvalidFormatException
    */
   public void copyPicture(XWPFRun target, XWPFPicture picture) throws IOException, InvalidFormatException {

      String filename = picture.getPictureData().getFileName();
      InputStream pictureData = new ByteArrayInputStream(picture
            .getPictureData().getData());
      int pictureType = picture.getPictureData().getPictureType();
      int width = (int) picture.getCTPicture().getSpPr().getXfrm().getExt()
            .getCx();

      int height = (int) picture.getCTPicture().getSpPr().getXfrm().getExt()
            .getCy();

      // target.addBreak();
      target.addPicture(pictureData, pictureType, filename, width, height);
      // target.addBreak(BreakType.PAGE);
   }

   /**
    * 复制段落，从source到target
    *
    * @param target
    * @param source
    */
   public void copyParagraph(XWPFParagraph target, XWPFParagraph source) {

      // 设置段落样式
      target.getCTP().setPPr(source.getCTP().getPPr());

      // 移除所有的run
      for (int pos = target.getRuns().size() - 1; pos >= 0; pos--) {
         target.removeRun(pos);
      }

      // copy 新的run
      for (XWPFRun s : source.getRuns()) {
         XWPFRun targetrun = target.createRun();
         copyRun(targetrun, s);
      }


   }

   /**
    * 复制RUN，从source到target
    *
    * @param target
    * @param source
    */
   public void copyRun(XWPFRun target, XWPFRun source) {
      // 设置run属性
      target.getCTR().setRPr(source.getCTR().getRPr());
      // 设置文本
      target.setText(source.text());
      // 处理图片
      List<XWPFPicture> pictures = source.getEmbeddedPictures();

      for (XWPFPicture picture : pictures) {
         try {
            copyPicture(target, picture);
         } catch (InvalidFormatException e) {
            logger.error("copyRun", e);
         } catch (IOException e) {
            logger.error("copyRun", e);
         }
      }
   }

}