jinlin
2024-01-02 a38a1ac77bb6ac9ea8bf0cf5f5f6b68d0e6e6974
core/src/main/java/com/zt/life/export/service/WordFileService.java
@@ -8,84 +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 {
    @Value("${zt.oss.local-path}")
    private String localPath;
    public void export(HttpServletRequest request, WordFile wordFile, HttpServletResponse response) throws IOException {
        Map<String, Object> datas = new HashMap<String, Object>();
        // 将word内容json字符串转成json对象
        Field[] fields = ReflectUtil.getFields(wordFile.getClass());
        for (Field field : fields) {
            try {
                field.setAccessible(true);
                //Object value = field.get(wordFile);
                datas.put(field.getName(), field.get(wordFile));
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
        }
        // 不支持doc的格式
        // 加载word导出模板,读入导出内容
        XWPFTemplate template = XWPFTemplate.compile(wordFile.getModulePath());
        template.render(datas);
        //FileOutputStream out = new FileOutputStream("D:\\out_template.docx");
        //template.write(out);
        //out.flush();
        //out.close();
        //template.close();
        DownloadService.download(request, response, template, wordFile.getWordName());
    }
    @Autowired
    private OssEncryptService ossEncryptService;
    /**
     * 打印word(单个表格数据)
     *
     * @param request
     * @param wordFile
     * @param response
     */
    public void exportForTables(HttpServletRequest request, WordFile wordFile, HttpServletResponse response) throws UnsupportedEncodingException, FileNotFoundException {
    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(dataObj, wordData, builder);
        Field[] fields = ReflectUtil.getFields(wordFile.getClass());
        for (Field field : fields) {
            field.setAccessible(true); // 对私有属性进行访问
            String fieldName = field.getName(); // 属性名
            Object staticFieldValue = ReflectUtil.getFieldValue(wordFile, field.getName()); // 属性对应的数据
            if (field.getType().getName().equals("java.util.List")) {//如果该属性是list类型
                builder.bind(fieldName, new HackLoopTableRenderPolicy());
            }else if(staticFieldValue!=null) {
                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")));
                     */
                }
            }
            data.put(fieldName, staticFieldValue);
            data.put(fieldName, staticFieldValue);
        }
        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) {
@@ -93,7 +49,57 @@
        } else {
            template = XWPFTemplate.compile(fs);
        }
        template.render(data);
        template.render(wordData);
        DownloadService.download(request, response, template, wordFile.getWordName());
    }
    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(dataObj, field.getName()); // 属性对应的数据
            if (typeName.contains("com.zt.life.modules")) {
                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 = 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();
                                if (width > 900 ) {
                                    height = (height * 900) / width;
                                    staticFieldValue = new PictureRenderData(900, height, ".png", in);
                                } else {
                                    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 (wordData.get(fieldName) == null)
                    wordData.put(fieldName, staticFieldValue);
            }
            if (wordData.get(fieldName) == null)
                wordData.put(fieldName, staticFieldValue);
        }
    }
}