From e4db7d4a3e03ec4767a57743aef01e54461bbe9a Mon Sep 17 00:00:00 2001
From: jinlin <jinlin>
Date: 星期一, 18 十二月 2023 14:29:11 +0800
Subject: [PATCH] 修改
---
core/src/main/java/com/zt/life/export/service/WordFileService.java | 119 ++++++++++++++++++++++++++++++-----------------------------
1 files changed, 60 insertions(+), 59 deletions(-)
diff --git a/core/src/main/java/com/zt/life/export/service/WordFileService.java b/core/src/main/java/com/zt/life/export/service/WordFileService.java
index 810b2a2..8b9208e 100644
--- a/core/src/main/java/com/zt/life/export/service/WordFileService.java
+++ b/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>();
- // 灏唚ord鍐呭json瀛楃涓茶浆鎴恓son瀵硅薄
-
- 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();
- }
- }
-
- // 涓嶆敮鎸乨oc鐨勬牸寮�
- // 鍔犺浇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,52 @@
} 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();
+ 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);
+ }
+ }
}
--
Gitblit v1.9.1