自学内容网 自学内容网

java 根据word模板,实现数据动态插入,包括二维码图片插入,并合并多个word文档,最终转为pdf导出

需求是要求查询数据库多条明细数据,将多条数据根据定好的word 模板,生成多个word文档(其中word文档中包含有二维码图片),并且需要将多个文档合并成一个文档,最终转换为pdf供前端导出和预览
我的实现逻辑如下:先根据查询的明细数据,在本地文件夹中生成多个word文档,并且将其一个个编号,同时每个文档对应的二维码图片也生成好,供后续插入word文档使用。然后分批次合并成一个word文档,最后转为pdf格式导出。为什么要分批次合并(如50个word合并一次),因为如果查询的明细条数太多的话,会造成阻塞。所以不仅要分批次合并,生成一个个word时还需要多线程执行。直接上代码:
首先是controller 层 没啥好说的

@ApiOperation("导出缴费单Pdf列表")
    @GetMapping("/exportwordPdf")
    public ResponseEntity<FileSystemResource> exportwordPdf(Long id, Date payDate, Date payDate2)
    {
        return paymentService.exportwordPdf(id,payDate,payDate2);
    }

重点说下impl实现层 注意 this.gettargetFiles(id,payDate,payDate2) 方法 这里是生成一个个word文档的方法,模板中以{{}} 作为占位符,然后一个个替换,我这里使用线程池实现多线程

@Value("${temp.outputPath}")
    private String outputPath; // 文档导出路径
    @Value("${temp.MODELPATH}")
    private String MODELPATH; // 模板文档临时存储路径
    @Value("${temp.qrcodeFont}") //二维码前缀
    private String qrcodeFont;
    private static final int BATCH_SIZE = 50;

@SneakyThrows
    @Override
    public ResponseEntity<FileSystemResource> exportwordPdf(Long id,Date payDate,Date payDate2) {
        List<File> targetFiles = this.gettargetFiles(id,payDate,payDate2);
        //合并文档
        if (targetFiles.size() > BATCH_SIZE) {
            return exportBatch(targetFiles, id, payDate, payDate2,true);
        } else {
            // 合并文档
            // 合并后文档路径
            File hbfile = new File(outputPath + File.separator+id+File.separator+ "output.docx");
            DocxMerge.appendDocx(hbfile, targetFiles);
            System.out.println("----合并成功-----");
            return createResponseEntity(hbfile,id,true);
        }
    }

@SneakyThrows
    private List<File> gettargetFiles (Long id, Date payDate, Date payDate2){
        //根据id查询缴费单详情列表
        FeeDetail feeDetail=new FeeDetail();
        feeDetail.setPaymentId(id);
        List<String> statusList = new ArrayList<>();
        statusList.add("1");
        statusList.add("2");
        feeDetail.setStatusList(statusList);
        List<FeeDetail> feeDetailslist = feeDetailService.selectFeeDetailList(feeDetail);
        // 假设这是从数据库查询出的多条记录,每条记录是一个Map
        List<Map<String, String>> allReplacements = new ArrayList<>();
        feeDetailslist.forEach(vo->{
            Map<String, String> record = new HashMap<>();
            record.put("{{start}}", DateUtils.parseDateToStr("yyyy年M月d日",vo.getStartDate()));
            record.put("{{end}}",  DateUtils.parseDateToStr("M月d日",vo.getEndDate()));
            record.put("{{shopNo}}", vo.getShopNo());
            record.put("{{feeDetailNo}}", vo.getFeeDetailNo());
            record.put("{{contractArea}}", "0");
            record.put("{{monthlyRent}}", "0");
            record.put("{{rent}}", vo.getRent()==null?"0":vo.getRent().stripTrailingZeros().toPlainString());
            record.put("{{waterAndEle}}", vo.getWaterAndElectricityFee()==null?"0":vo.getWaterAndElectricityFee().stripTrailingZeros().toPlainString());
            record.put("{{airCond}}", vo.getAirConditioningFee()==null?"0":vo.getAirConditioningFee().stripTrailingZeros().toPlainString());
            record.put("{{adFee}}", vo.getAdvertisingFee()==null?"0":vo.getAdvertisingFee().stripTrailingZeros().toPlainString());
            record.put("{{manageFee}}", vo.getManagementFee()==null?"0":vo.getManagementFee().stripTrailingZeros().toPlainString());
            record.put("{{lampFee}}", vo.getLampFee()==null?"0":vo.getLampFee().stripTrailingZeros().toPlainString());
            record.put("{{totalFee}}", vo.getTotalFee()==null?"0":vo.getTotalFee().stripTrailingZeros().toPlainString());
            record.put("{{otherFee}}", vo.getPropertyManagementFee()==null?"0":vo.getPropertyManagementFee().stripTrailingZeros().toPlainString());  //其他费用 物管费
            record.put("{{totalFeebig}}", Convert.digitToChinese(vo.getTotalFee()));
            record.put("{{startdate}}", DateUtils.parseDateToStr("yyyy.M.d",vo.getStartDate()));
            record.put("{{enddate}}", DateUtils.parseDateToStr("yyyy.M.d",vo.getEndDate()));
            record.put("{{paydate}}", DateUtils.parseDateToStr("M月d日",payDate));
            record.put("{{paydate2}}", DateUtils.parseDateToStr("M月d日",payDate2));
            allReplacements.add(record);
        });

        String modelPath =MODELPATH;
        //导出目录不存在则新建
        File folder = new File(outputPath+ File.separator +id);
        if(!folder.exists()){
            folder.mkdirs();
        }
        //需要合并的文档集合
        List<File> targetFiles = new ArrayList<>();
        ExecutorService executor = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
        List<Future<File>> futures = new ArrayList<>();
        for (int i = 0; i < allReplacements.size(); i++) {
            int index = i; // 将 i 的值存储在一个局部变量中
            Map<String, String> replacement = allReplacements.get(i); // 将获取的 Map 存储在局部变量中
            futures.add(executor.submit(() -> {
                XWPFDocument doc = new XWPFDocument(new FileInputStream(modelPath));
                WordWithQRCode.replaceInParagraphs(doc, replacement); //替换占位符
                //生成二维码图片
                String value=replacement.get("{{feeDetailNo}}");
                String params="{\"input_1717459797397\":\""+ value + "\"}";
                String QRCODE=qrcodeFont+ Base64Utilt.encoder64(params)  ;
                String codePath = outputPath + File.separator +id +File.separator+ index + ".png";
                WordWithQRCode.generateQRCode(QRCODE, codePath, 150, 150);
                Map<String, String> imageMap = new HashMap<>();
                imageMap.put("{{qrcode}}", codePath);
                WordWithQRCode.replaceImages(doc, imageMap);
                String outpath = outputPath + File.separator +id +File.separator+ index + ".docx";
                FileOutputStream out = new FileOutputStream(outpath);
                doc.write(out);
                return new File(outpath);
            }));
        }
        // 等待所有文档生成完成
        for (Future<File> future : futures) {
            targetFiles.add(future.get());
        }
        // 关闭线程池
        executor.shutdown();
        System.out.print("----word生成成功-----");
        return targetFiles;
    }

其中 WordWithQRCode 是生成word的主要方法,我将其写成了一个工具类, WordWithQRCode.replaceInParagraphs 方法是替换模板占位符生成word文档,
WordWithQRCode.generateQRCode 方法是生成二维码图片的功能,
WordWithQRCode.replaceImages 方法是将生成的二维码图片插入文档占位符的方法,
我的模板中涉及到金额的中文大小写转换的功能,你不需要就删掉
全部代码如下:

package com.nrx.contract.utils;

import com.fasterxml.jackson.databind.exc.InvalidFormatException;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.nrx.common.utils.StringUtils;
import lombok.SneakyThrows;
import org.apache.poi.util.Units;
import org.apache.poi.xwpf.usermodel.*;

import java.io.*;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.*;

/**
 * 根据word模板生成 word文档
 * */
public class WordWithQRCode {

    public static void main(String[] args) throws IOException {
        String templatePath = "C:\\Users\\hc\\Desktop\\bushu\\model\\jfd.docx"; // 模板文件路径
        String outputPath = "C:\\Users\\hc\\Desktop\\tx"; // 输出文件路径
        String qrcode="哈哈哈哈,我是二维码";


        // 假设这是从数据库查询出的多条记录,每条记录是一个Map
        List<Map<String, String>> allReplacements = new ArrayList<>();
        Map<String, String> record1 = new HashMap<>();
        record1.put("{{start}}", "2024-05-01");
        record1.put("{{end}}", "2024-05-22");
        record1.put("{{shopNo}}", "001");
        record1.put("{{feeDetailNo}}", "No111");
        record1.put("{{contractArea}}", "111");
        record1.put("{{monthlyRent}}", "211");
        record1.put("{{rent}}", "311");
        record1.put("{{waterAndEle}}", "411");
        record1.put("{{airCond}}", "511");
        record1.put("{{adFee}}", "611");
        record1.put("{{manageFee}}", "711");
        record1.put("{{lampFee}}", "811");
        record1.put("{{totalFee}}", "911");
        record1.put("{{otherFee}}", "0");
        record1.put("{{totalFeebig}}", toChinese("911", true));
        record1.put("{{startdate}}", "2024.05.01");
        record1.put("{{enddate}}", "2024.05.22");
        allReplacements.add(record1);

        Map<String, String> record2 = new HashMap<>();
        record2.put("{{start}}", "2024-05-02");
        record2.put("{{end}}", "2024-05-23");
        record2.put("{{shopNo}}", "002");
        record2.put("{{feeDetailNo}}", "No222");
        record2.put("{{contractArea}}", "112");
        record2.put("{{monthlyRent}}", "212");
        record2.put("{{rent}}", "312");
        record2.put("{{waterAndEle}}", "412");
        record2.put("{{airCond}}", "512");
        record2.put("{{adFee}}", "612");
        record2.put("{{manageFee}}", "712");
        record2.put("{{lampFee}}", "812");
        record2.put("{{totalFee}}", "912");
        record2.put("{{otherFee}}", "0");
        record2.put("{{totalFeebig}}", toChinese("911",true));
        record2.put("{{startdate}}", "2024.05.02");
        record2.put("{{enddate}}", "2024.05.23");
        allReplacements.add(record2);

        //需要合并的文档集合
        List<File> targetFile = new ArrayList<>();
        //生成每个word
        for (int i=0;i<allReplacements.size();i++){
            XWPFDocument doc = new XWPFDocument(new FileInputStream(templatePath));
            // 替换段落
            replaceInParagraphs(doc, allReplacements.get(i));

            //生成二维码
            String codePath=outputPath+File.separator+i+".png";
            generateQRCode(qrcode,codePath,100,100);
            //如果有图片占位符
            Map<String, String> imageMap = new HashMap<>();
            imageMap.put("{{qrcode}}", codePath);
            replaceImages(doc, imageMap);

            //文档输出路径
            String outpath=outputPath+File.separator+i+".docx";
            FileOutputStream out = new FileOutputStream(outpath);
            doc.write(out);
            targetFile.add(new File(outpath));
        }
        System.out.printf("----生成成功-----");
        //合并文档
        //合并后文档路径
        File hbfile = new File(outputPath+File.separator+"output.docx");
        DocxMerge.appendDocx(hbfile, targetFile);
        System.out.printf("----合并成功-----");
        //转pdf文档
        Word2PdfUtil.wordConvertPdfFile(hbfile.getPath(),outputPath+File.separator+"output.pdf");

        System.out.println("Word文档已成功转换为PDF格式。");
    }


    //生成二维码
    public static void generateQRCode(String text, String filePath, int width, int height) {
        try {
            BitMatrix bitMatrix = new MultiFormatWriter().encode(text, BarcodeFormat.QR_CODE, width, height);

            Path path = FileSystems.getDefault().getPath(filePath);
            MatrixToImageWriter.writeToPath(bitMatrix, "PNG", path);
        } catch (WriterException | IOException e) {
            e.printStackTrace();
        }
    }





    public static void replaceInParagraphs(XWPFDocument doc, Map<String, String> params) {
        // 替换文档中的所有段落
        replaceInParagraphs(doc.getParagraphs(), params);
        // 替换表格中的内容(如果有表格)
        replaceInTables(doc.getTables(), params);
        // 替换页眉中的内容(如果有页眉)
        replaceInHeaders(doc.getHeaderList(), params);
        // 替换页脚中的内容(如果有页脚)
        replaceInFooters(doc.getFooterList(), params);
    }
    private static void replaceInParagraphs(List<XWPFParagraph> paragraphs, Map<String, String> replacements) {
        for (XWPFParagraph paragraph : paragraphs) {
            replaceText(paragraph, replacements);
        }
    }

    private static void replaceInTables(List<XWPFTable> tables, Map<String, String> replacements) {
        for (XWPFTable table : tables) {
            for (XWPFTableRow row : table.getRows()) {
                for (XWPFTableCell cell : row.getTableCells()) {
                    replaceInParagraphs(cell.getParagraphs(), replacements);
                }
            }
        }
    }

    private static void replaceInHeaders(List<XWPFHeader> headers, Map<String, String> replacements) {
        for (XWPFHeader header : headers) {
            replaceInParagraphs(header.getParagraphs(), replacements);
        }
    }

    private static void replaceInFooters(List<XWPFFooter> footers, Map<String, String> replacements) {
        for (XWPFFooter footer : footers) {
            replaceInParagraphs(footer.getParagraphs(), replacements);
        }
    }

    private static void replaceText(XWPFParagraph paragraph, Map<String, String> replacements) {
        StringBuilder fullText = new StringBuilder();
        List<XWPFRun> runs = paragraph.getRuns();

        if (runs != null) {
            for (XWPFRun run : runs) {
                String text = run.getText(0);
                if (text != null) {
                    fullText.append(text);
                }
            }

            // 替换文本中的占位符
            String replacedText = fullText.toString();
            for (Map.Entry<String, String> entry : replacements.entrySet()) {
                replacedText = replacedText.replace(entry.getKey(), entry.getValue());
            }

            // 设置替换后的文本
            for (XWPFRun run : runs) {
                run.setText("", 0); // 清空原有文本
            }
            if (runs.size() > 0) {
                runs.get(0).setText(replacedText, 0);
            }
        }
    }
    @SneakyThrows
    public static void replaceImages(XWPFDocument document, Map<String, String> imageMap) throws IOException {
        for (XWPFTable table : document.getTables()) {
            for (XWPFTableRow row : table.getRows()) {
                for (XWPFTableCell cell : row.getTableCells()) {
                    for (XWPFParagraph paragraph : cell.getParagraphs()) {
                        for (XWPFRun run : paragraph.getRuns()) {
                            String text = run.getText(0);
                            if (text != null) {
                                for (Map.Entry<String, String> entry : imageMap.entrySet()) {
                                    String placeholder = entry.getKey();
                                    String imagePath = entry.getValue();

                                    if (text.contains(placeholder)) {
                                        // Replace placeholder text with empty string
                                        run.setText(text.replace(placeholder, ""), 0);

                                        // Load and add image
                                        byte[] imageBytes = Files.readAllBytes(Paths.get(imagePath));
                                        try {
                                            String pictureIndex = document.addPictureData(imageBytes, 5);
                                            // 下面参数5是图片类型,这里是png对应的数字,如果是其它的可以自行百度,
                                            // 200设置的是图片高度,400是图片宽度
                                            run.addPicture(new ByteArrayInputStream(imageBytes), 5, "image.png", Units.toEMU(100), Units.toEMU(100));
                                        } catch (InvalidFormatException e) {
                                            throw new RuntimeException(e);
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            }
        }
    }


    /**
     * 繁体大写数字************************************************************
     */
    private static final String[] NUMBERS = {"零", "壹", "贰", "叁", "肆", "伍", "陆", "柒", "捌", "玖"};
    /**
     * 繁体整数部分的单位
     */
    private static final String[] IUNIT = {"元", "拾", "佰", "仟", "万", "拾", "佰", "仟", "亿", "拾", "佰", "仟"};
    /**
     * 繁体小数部分的单位
     */
    private static final String[] DUNIT = {"角", "分"};
    /**
     * 简体数字
     */
    private static final String[] CN_NUMBERS = {"零", "一", "二", "三", "四", "五", "六", "七", "八", "九"};
    /**
     * 简体数字单位
     */
    private static final String[] CN_IUNIT = {"", "十", "百", "千", "万", "十", "百", "千", "亿", "十", "百", "千"};


    /**
     *  转换为大写的中文金额,支持负数
     * @param amount 金额
     * @param isSimplified 是否简体中文:true:简体,false:繁体
     * @return
     */
    public static String toChinese(String amount, boolean isSimplified) {
        // 判断输入的金额字符串是否符合要求
        if (StringUtils.isBlank(amount) || !amount.matches("(-)?[\\d]*(.)?[\\d]*")) {
            throw new RuntimeException("请输入数字");
        }

        if ("0".equals(amount) || "0.00".equals(amount) || "0.0".equals(amount)) {
            return isSimplified ? "零" : "零元";
        }

        // 判断金额数字中是否存在负号"-"
        boolean flag = false;
        if (amount.startsWith("-")) {
            // 标志位,标志此金额数字为负数
            flag = true;
            amount = amount.replaceAll("-", "");
        }
        // 去掉金额数字中的逗号","
        amount = amount.replaceAll(",", "");
        // 初始化:分离整数部分和小数部分
        String[] separateNum = separateNum(amount);
        // 整数部分数字
        String integerStr = separateNum[0];
        // 小数部分数字
        String decimalStr = separateNum[1];
        // beyond超出计算能力,直接返回
        if (integerStr.length() > IUNIT.length) {
            throw new RuntimeException("输入数字超限");
        }
        // 整数部分数字
        int[] integers = toIntArray(integerStr);
        // 判断整数部分是否存在输入012的情况
        if (integers.length > 1 && integers[0] == 0) {
            throw new RuntimeException("输入数字不符合要求");
        }
        // 设置万单位
        boolean isWan = isWan5(integerStr);
        // 小数部分数字
        int[] decimals = toIntArray(decimalStr);
        // 返回最终的大写金额
        String result = "";
        String chineseInteger = getChineseInteger(integers, isWan, isSimplified);
        String chineseDecimal = getChineseDecimal(decimals, isSimplified);
        if (decimals.length > 0 && isSimplified) {
            result = chineseInteger;
            if (!chineseDecimal.equals("零零")) {
                result = result + "点" + chineseDecimal;
            }
        } else {
            result = chineseInteger + chineseDecimal;

        }
        if (flag) {
            // 如果是负数,加上"负"
            return "负" + result;
        } else {
            return result;
        }
    }

    /**
     * 分离整数部分和小数部分
     * @param str
     * @return
     */
    private static String[] separateNum(String str) {
        String integerStr;// 整数部分数字
        String decimalStr;// 小数部分数字
        if (str.indexOf('.') >= 1) {
            integerStr = str.substring(0, str.indexOf('.'));
            decimalStr = str.substring(str.indexOf('.') + 1);
            if (decimalStr.length() > 2) {
                decimalStr = decimalStr.substring(0, 2);
            }
        } else if (str.indexOf('.') == 0) {
            integerStr = "";
            decimalStr = str.substring(1);
        } else {
            integerStr = str;
            decimalStr = "";
        }
        return new String[] {integerStr, decimalStr};
    }

    /**
     *  将字符串转为int数组
     * @param number  数字
     * @return
     */
    private static int[] toIntArray(String number) {
        int[] array = new int[number.length()];
        for (int i = 0; i < number.length(); i++) {
            array[i] = Integer.parseInt(number.substring(i, i + 1));
        }
        return array;
    }

    /**
     *  将整数部分转为大写的金额
     * @param integers 整数部分数字
     * @param isWan  整数部分是否已经是达到【万】
     * @return
     */
    private static String getChineseInteger(int[] integers, boolean isWan, boolean isSimplified) {

        int length = integers.length;
        if (!isSimplified && length == 1 && integers[0] == 0) {
            return "";
        }
        if (!isSimplified) {
            return traditionalChineseInteger(integers, isWan);
        } else {
            return simplifiedChineseInteger(integers, isWan);
        }
    }

    /**
     * 繁体中文整数
     * @param integers
     * @param isWan
     * @return
     */
    private static String traditionalChineseInteger(int[] integers, boolean isWan) {
        StringBuilder chineseInteger = new StringBuilder("");
        int length = integers.length;
        for (int i = 0; i < length; i++) {
            String key = "";
            if (integers[i] == 0) {
                if ((length - i) == 13)// 万(亿)
                    key = IUNIT[4];
                else if ((length - i) == 9) {// 亿
                    key = IUNIT[8];
                } else if ((length - i) == 5 && isWan) {// 万
                    key = IUNIT[4];
                } else if ((length - i) == 1) {// 元
                    key = IUNIT[0];
                }
                if ((length - i) > 1 && integers[i + 1] != 0) {
                    key += NUMBERS[0];
                }
            }
            chineseInteger.append(integers[i] == 0 ? key : (NUMBERS[integers[i]] + IUNIT[length - i - 1]));
        }
        return chineseInteger.toString();
    }

    /**
     * 简体中文整数
     * @param integers
     * @param isWan
     * @return
     */
    private static String simplifiedChineseInteger(int[] integers, boolean isWan) {
        StringBuilder chineseInteger = new StringBuilder("");
        int length = integers.length;
        for (int i = 0; i < length; i++) {
            String key = "";
            if (integers[i] == 0) {
                if ((length - i) == 13) {// 万(亿)
                    key = CN_IUNIT[4];
                } else if ((length - i) == 9) {// 亿
                    key = CN_IUNIT[8];
                } else if ((length - i) == 5 && isWan) {// 万
                    key = CN_IUNIT[4];
                } else if ((length - i) == 1) {// 元
                    key = CN_IUNIT[0];
                }
                if ((length - i) > 1 && integers[i + 1] != 0) {
                    key += CN_NUMBERS[0];
                }
                if (length == 1 && integers[i] == 0) {
                    key += CN_NUMBERS[0];
                }
            }
            chineseInteger.append(integers[i] == 0 ? key : (CN_NUMBERS[integers[i]] + CN_IUNIT[length - i - 1]));
        }
        return chineseInteger.toString();
    }

    /**
     *  将小数部分转为大写的金额
     * @param decimals 小数部分的数字
     * @return
     */
    private static String getChineseDecimal(int[] decimals, boolean isSimplified) {
        StringBuilder chineseDecimal = new StringBuilder("");
        if (!isSimplified) {
            for (int i = 0; i < decimals.length; i++) {
                String key = "";

                if ((decimals.length - i) > 1 && decimals[i + 1] != 0) {
                    key += NUMBERS[0];
                }

                chineseDecimal.append(decimals[i] == 0 ? key : (NUMBERS[decimals[i]] + DUNIT[i]));
            }
        } else {
            for (int i = 0; i < decimals.length; i++) {
                chineseDecimal.append(CN_NUMBERS[decimals[i]]);
            }

        }
        return chineseDecimal.toString();
    }

    /**
     *  判断当前整数部分是否已经是达到【万】
     * @param integerStr  整数部分数字
     * @return
     */
    private static boolean isWan5(String integerStr) {
        int length = integerStr.length();
        if (length > 4) {
            String subInteger = "";
            if (length > 8) {
                subInteger = integerStr.substring(length - 8, length - 4);
            } else {
                subInteger = integerStr.substring(0, length - 4);
            }
            return Integer.parseInt(subInteger) > 0;
        } else {
            return false;
        }
    }
    /**
     *************************************************************
     */

    /**
     * 删除文件夹下文件
     * */
    public static void deleteFile(String path) {
        File directory = new File("path");
        if(directory.listFiles()!=null){
            for (File file: directory.listFiles()) {
                if (!file.isDirectory()) {
                    file.delete();
                }
            }
        }

    }

}

简单说一下exportBatch() 方法,此方法是用于分批次合并word文档的里面的DocxMerge.appendDocx() 才是真正合并word文档的方法,阈值为BATCH_SIZE 如果大于则分批次,不大于则直接执行DocxMerge.appendDocx()方法 如下:

 @SneakyThrows
    private ResponseEntity<FileSystemResource> exportBatch(List<File> targetFiles, Long id, Date payDate, Date payDate2, boolean changePdf) {
        System.out.println("----大于"+BATCH_SIZE+"分批次合并中-----");
        List<List<File>> batches = new ArrayList<>();
        for (int i = 0; i < targetFiles.size(); i += BATCH_SIZE) {
            batches.add(new ArrayList<>(targetFiles.subList(i, Math.min(i + BATCH_SIZE, targetFiles.size()))));
        }

        File finalMergedFile = new File(outputPath + File.separator +id+File.separator+ "final_output.docx");

        List<File> allIntermediateFiles = new ArrayList<>(); // 用于存储所有中间文件

        // 创建第一个中间文件,并将其作为临时最终文件的基础
        List<File> firstBatch = batches.get(0);
        File tempFinalMergedFile = new File(outputPath + File.separator +id+File.separator+ "intermediate_0.docx");
        DocxMerge.appendDocx(tempFinalMergedFile, firstBatch);
        allIntermediateFiles.add(tempFinalMergedFile);

        // 合并剩余的中间文件
        for (int i = 1; i < batches.size(); i++) {
            List<File> batch = batches.get(i);
            File intermediateMergedFile = new File(outputPath + File.separator +id+File.separator+ "intermediate_" + i + ".docx");
            DocxMerge.appendDocx(intermediateMergedFile, batch);
            allIntermediateFiles.add(intermediateMergedFile);
        }

        // 最终合并所有中间文件到最终文件
        DocxMerge.appendDocx(finalMergedFile, allIntermediateFiles);

        // 清理中间文件
        for (File file : allIntermediateFiles) {
            file.delete();
        }

        System.out.println("----分批次合并成功-----");
        return createResponseEntity(finalMergedFile,id, changePdf);
    }

 private ResponseEntity<FileSystemResource> createResponseEntity(File file,Long id,boolean changepdf) {
        HttpHeaders headers = new HttpHeaders();
        headers.add("Cache-Control", "no-cache, no-store, must-revalidate");
        headers.add("Pragma", "no-cache");
        headers.add("Expires", "0");
        headers.add("Last-Modified", new Date().toString());
        headers.add("ETag", String.valueOf(System.currentTimeMillis()));

        if(changepdf){
            //转换为pdf
            File hbfilepdf = new File(outputPath+File.separator+id+File.separator+"output.pdf");
            Word2PdfUtil.wordConvertPdfFile(file.getPath(),hbfilepdf.getPath());
            System.out.println("----转换为pdf成功-----");
            headers.add("Content-Disposition", "attachment; filename=" + UriUtils.encode(hbfilepdf.getName(), "UTF-8"));
            return ResponseEntity
                    .ok()
                    .headers(headers)
                    .contentLength(hbfilepdf.length())
                    .contentType(MediaType.parseMediaType("application/octet-stream"))
                    .body(new FileSystemResource(hbfilepdf));
        }else{
            headers.add("Content-Disposition", "attachment; filename=" + UriUtils.encode(file.getName(), "UTF-8"));
            return ResponseEntity
                    .ok()
                    .headers(headers)
                    .contentLength(file.length())
                    .contentType(MediaType.parseMediaType("application/octet-stream"))
                    .body(new FileSystemResource(file));
        }
    }

DocxMerge.appendDocx 合并word文档的工具方法:此方法很重要,代码如下:

package com.nrx.contract.utils;
import lombok.SneakyThrows;
import org.apache.poi.openxml4j.util.ZipSecureFile;
import org.apache.poi.xwpf.usermodel.*;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.xmlbeans.XmlObject;
import org.apache.xmlbeans.XmlOptions;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTBody;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTP;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTR;

import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.*;

/**
 * @Author mischen
 * @Description 文件合并 只支持.docx
 * @Date 2022/11/21 16:30
 * @Version 1.0
 */
public class DocxMerge {
    public static void main(String[] args) {
        File file1 = new File("C:\\Users\\hc\\Desktop\\tx\\output.docx");
        List<File> targetFile1 = new ArrayList<>();
        targetFile1.add(new File("C:\\Users\\hc\\Desktop\\tx\\0.docx"));
        targetFile1.add(new File("C:\\Users\\hc\\Desktop\\tx\\1.docx"));

        appendDocx(file1, targetFile1);
        System.out.println("合并成功!!!");
    }

    /**
     * 把多个docx文件合并成一个
     *
     * @param outfile    输出文件
     * @param targetFile 目标文件
     */
   
    /**
     * 把多个docx文件合并成一个
     *
     * @param outfile    输出文件
     * @param targetFile 目标文件
     */
   public static void appendDocx(File outfile, List<File> targetFile) {
        try {

            OutputStream dest = new FileOutputStream(outfile);
            ArrayList<XWPFDocument> documentList = new ArrayList<>();

            for (int i = 0; i < targetFile.size(); i++) {
                ZipSecureFile.setMinInflateRatio(-1.0d);
                FileInputStream in = new FileInputStream(targetFile.get(i).getPath());
                OPCPackage open = OPCPackage.open(in);
                XWPFDocument document = new XWPFDocument(open);
                documentList.add(document);
                in.close();
            }
            //取出第一个用作基础
            XWPFDocument doc =new XWPFDocument();
            if(!documentList.isEmpty()){
                doc= documentList.get(0);
                for (int i = 1; i < documentList.size(); i++) {
                    //解决word合并完后,所有表格都紧紧挨在一起,没有分页。加上了分页符可解决
                    //  insertPageBreak(documentList.get(i));
                    //在第一个后面追加
                    appendBody(doc, documentList.get(i));
                    System.out.println("----合并中---已处理"+i);
                }
            }

            doc.write(dest);
            dest.close();
            doc.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private static void insertPageBreak(XWPFDocument document) {
        XWPFParagraph paragraph = document.createParagraph();
        XWPFRun run = paragraph.createRun();
        run.addBreak(BreakType.PAGE);
    }


    public static void appendBody(XWPFDocument src, XWPFDocument append) throws Exception {
        CTBody src1Body = src.getDocument().getBody();
        CTBody src2Body = append.getDocument().getBody();

        List<XWPFPictureData> allPictures = append.getAllPictures();
        // 记录图片合并前及合并后的ID
        Map<String, String> map = new HashMap<>();
        for (XWPFPictureData picture : allPictures) {
            String before = append.getRelationId(picture);
            //将原文档中的图片加入到目标文档中
            String after = src.addPictureData(picture.getData(), Document.PICTURE_TYPE_PNG);
            map.put(before, after);
        }
        //这个代码主要解决合并word报错,解析抛出压缩炸弹
        ZipSecureFile.setMinInflateRatio(-1.0d);
        appendBody(src1Body, src2Body, map);

    }

    private static void appendBody(CTBody src, CTBody append, Map<String, String> map) throws Exception {
        XmlOptions optionsOuter = new XmlOptions();
        optionsOuter.setSaveOuter();
        String appendString = append.xmlText(optionsOuter);

        String srcString = src.xmlText();
        String prefix = srcString.substring(0, srcString.indexOf(">") + 1);
        String mainPart = srcString.substring(srcString.indexOf(">") + 1, srcString.lastIndexOf("<"));
        String sufix = srcString.substring(srcString.lastIndexOf("<"));
        String addPart = appendString.substring(appendString.indexOf(">") + 1, appendString.lastIndexOf("<"));
        //下面这部分可以去掉,我加上的原因是合并的时候,有时候出现打不开的情况,对照document.xml将某些标签去掉就可以正常打开了
        addPart = addPart.replaceAll("w14:paraId=\"[A-Za-z0-9]{1,10}\"", "");
        addPart = addPart.replaceAll("w14:textId=\"[A-Za-z0-9]{1,10}\"", "");
        addPart = addPart.replaceAll("w:rsidP=\"[A-Za-z0-9]{1,10}\"", "");
        addPart = addPart.replaceAll("w:rsidRPr=\"[A-Za-z0-9]{1,10}\"", "");
        addPart = addPart.replace("<w:headerReference r:id=\"rId8\" w:type=\"default\"/>","");
        addPart = addPart.replace("<w:footerReference r:id=\"rId9\" w:type=\"default\"/>","");
        addPart = addPart.replace("xsi:nil=\"true\"","");

        if (map != null && !map.isEmpty()) {
            //对xml字符串中图片ID进行替换
            for (Map.Entry<String, String> set : map.entrySet()) {
                addPart = addPart.replace(set.getKey(), set.getValue());
            }
        }
        //将两个文档的xml内容进行拼接
        XmlObject makeBody = CTBody.Factory.parse(prefix + mainPart + addPart + sufix);

        src.set(makeBody);
    }
}

Word2PdfUtil.wordConvertPdfFile 是word转换为pdf文档的工具类方法

package com.nrx.contract.utils;

import com.aspose.words.Document;
import com.aspose.words.License;
import com.aspose.words.SaveFormat;

import java.io.*;

/**
 * word转pdf工具类
 *
 * @author shmily
 */
public class Word2PdfUtil {

    /**
     * 许可证字符串(可以放到resource下的xml文件中也可)
     */
    private static final String LICENSE = "<License>" +
            "<Data>" +
            "<Products><Product>Aspose.Total for Java</Product><Product>Aspose.Words for Java</Product></Products>" +
            "<EditionType>Enterprise</EditionType>" +
            "<SubscriptionExpiry>20991231</SubscriptionExpiry>" +
            "<LicenseExpiry>20991231</LicenseExpiry>" +
            "<SerialNumber>8bfe198c-7f0c-4ef8-8ff0-acc3237bf0d7</SerialNumber>" +
            "</Data>" +
            "<Signature>sNLLKGMUdF0r8O1kKilWAGdgfs2BvJb/2Xp8p5iuDVfZXmhppo+d0Ran1P9TKdjV4ABwAgKXxJ3jcQTqE/2IRfqwnPf8itN8aFZlV3TJPYeD3yWE7IT55Gz6EijUpC7aKeoohTb4w2fpox58wWoF3SNp6sK6jDfiAUGEHYJ9pjU=</Signature>" +
            "</License>";


    /**
     * 设置 license 去除水印
     */
    private static void setLicense() {
        ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(LICENSE.getBytes());
        License license = new License();
        try {
            license.setLicense(byteArrayInputStream);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }


    /**
     * word 转 pdf 生成至指定路径,pdf为空则上传至word同级目录
     *
     * @param wordPath word文件路径
     * @param pdfPath  pdf文件路径
     */
    public static void wordConvertPdfFile(String wordPath, String pdfPath) {
        FileOutputStream fileOutputStream = null;
        try {
            pdfPath = pdfPath == null ? getPdfFilePath(wordPath) : pdfPath;
            setLicense();
            File file = new File(pdfPath);
            fileOutputStream = new FileOutputStream(file);
            Document doc = new Document(wordPath);
            doc.save(fileOutputStream, SaveFormat.PDF);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                assert fileOutputStream != null;
                fileOutputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }

        }
    }


    /**
     * word 转 pdf 生成byte字节流
     *
     * @param wordPath word所在的目录地址
     * @return
     */
    public static byte[] wordConvertPdfByte(String wordPath) {
        ByteArrayOutputStream fileOutputStream = null;
        try {
            setLicense();
            fileOutputStream = new ByteArrayOutputStream();
            Document doc = new Document(wordPath);
            doc.save(fileOutputStream, SaveFormat.PDF);
            return fileOutputStream.toByteArray();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                assert fileOutputStream != null;
                fileOutputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }

        }
        return null;
    }



    /**
     * 获取 生成的 pdf 文件路径,默认与源文件同一目录
     *
     * @param wordPath word文件
     * @return 生成的 pdf 文件
     */
    private static String getPdfFilePath(String wordPath) {
        int lastIndexOfPoint = wordPath.lastIndexOf(".");
        String pdfFilePath = "";
        if (lastIndexOfPoint > -1) {
            pdfFilePath = wordPath.substring(0, lastIndexOfPoint);
        }
        return pdfFilePath + ".pdf";
    }

}

后面就不一一说明了,大家可以直接看资源:https://download.csdn.net/download/weixin_43832166/89893033


原文地址:https://blog.csdn.net/weixin_43832166/article/details/142210376

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!