修改

jinlin

2023-12-18 e4db7d4a3e03ec4767a57743aef01e54461bbe9a

 core/src/main/java/com/zt/life/export/service/WordFileService.java

@@ -8,27 +8,40 @@
import com.deepoove.poi.policy.HackLoopTableRenderPolicy;

import com.zt.life.export.dto.WordFile;
import com.zt.life.oss.OssEncryptService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;


import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.awt.image.BufferedImage;
import java.io.*;
import java.lang.reflect.Field;
import java.text.SimpleDateFormat;
import java.util.*;

@Service
public class WordFileService {
    public void exportWordFile(HttpServletRequest request, WordFile wordFile, HttpServletResponse response) throws UnsupportedEncodingException, FileNotFoundException {
    @Value("${zt.oss.local-path}")
    private String localPath;

    @Autowired
    private OssEncryptService ossEncryptService;


    public void exportWordFile(HttpServletRequest request, Object dataObj, WordFile wordFile, HttpServletResponse response) throws UnsupportedEncodingException, FileNotFoundException {
        // 数据map集合
        Map<String, Object> data = new HashMap<>();
        Map<String, Object> wordData = new HashMap<>();
        //HackLoopTableRenderPolicy hackLoopTableRenderPolicy = new HackLoopTableRenderPolicy();
        Configure config = null;
        ConfigureBuilder builder = Configure.newBuilder();
        this.formatWordData(wordFile, data, builder);
        this.formatWordData(dataObj, wordData, builder);

        config = builder.build();
        File fl = new File(wordFile.getModulePath());
        File fl = new File(localPath + "/template/" + wordFile.getModulePath());
        FileInputStream fs = new FileInputStream(fl);
        XWPFTemplate template;
        if (config != null) {
@@ -36,33 +49,52 @@
        } else {
            template = XWPFTemplate.compile(fs);
        }
        template.render(data);
        template.render(wordData);
        DownloadService.download(request, response, template, wordFile.getWordName());
    }

    public void formatWordData(Object obj, Map<String, Object> data, ConfigureBuilder builder) {
        Field[] fields = ReflectUtil.getFields(obj.getClass());
    public void formatWordData(Object dataObj, Map<String, Object> wordData, ConfigureBuilder builder) {
        Field[] fields = ReflectUtil.getFields(dataObj.getClass());
        for (Field field : fields) {
            field.setAccessible(true);
            String fieldName = field.getName();
            String typeName = field.getType().getName();
            Object staticFieldValue = ReflectUtil.getFieldValue(obj, field.getName()); // 属性对应的数据
            Object staticFieldValue = ReflectUtil.getFieldValue(dataObj, field.getName()); // 属性对应的数据
            if (typeName.contains("com.zt.life.modules")) {
                formatWordData(staticFieldValue, data, builder);
                formatWordData(staticFieldValue, wordData, builder);
            } else {
                if (field.getType().getName().equals("java.util.List")) {
                    builder.bind(fieldName, new HackLoopTableRenderPolicy());
                } else if (staticFieldValue != null) {
                    if (typeName.contains("java.util.Date") && fieldName.contains("Date")) {
                        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
                        staticFieldValue = sdf.format(staticFieldValue);
                    }
                    String valStr = staticFieldValue.toString();
                    if (valStr.contains("签名图片:")) {
                        valStr = valStr.replace("签名图片:", "");
                        staticFieldValue = new PictureRenderData(80, 100, "d://" + valStr);
                        /*put("localbyte", new PictureRenderData(80, 100, ".png", new FileInputStream("./logo.png")));*/
                    if (valStr.contains("文件图片:")) {
                        valStr = localPath + valStr.replace("文件图片:", "");
                        File file = new File(valStr);
                        if (file.exists() && !localPath.equals(valStr)) {
                            try (InputStream in = ossEncryptService.decrypt(file)) {
                                InputStream in2 = ossEncryptService.decrypt(file);
                                BufferedImage bufferedImage = ImageIO.read(in2);
                                int width = bufferedImage.getWidth();
                                int height = bufferedImage.getHeight();
                                staticFieldValue = new PictureRenderData(width, height, ".png", in);
                                /*staticFieldValue = new PictureRenderData(80, 100, "d://" + valStr);*/
                            } catch (IOException e) {
                                e.printStackTrace();
                            }
                        } else {
                            staticFieldValue = new PictureRenderData(100, 30, localPath+"template/noSign.png");
                        }
                    }
                }
                if (data.get(fieldName) == null)
                    data.put(fieldName, staticFieldValue);
                if (wordData.get(fieldName) == null)
                    wordData.put(fieldName, staticFieldValue);
            }
            if (wordData.get(fieldName) == null)
                wordData.put(fieldName, staticFieldValue);
        }
    }
}