package com.zt.life.export.service;
|
|
import cn.hutool.core.util.ReflectUtil;
|
import com.deepoove.poi.XWPFTemplate;
|
import com.deepoove.poi.config.Configure;
|
import com.deepoove.poi.config.ConfigureBuilder;
|
import com.deepoove.poi.data.PictureRenderData;
|
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;
|
|
@Autowired
|
private OssEncryptService ossEncryptService;
|
|
|
public void exportWordFile(HttpServletRequest request, Object dataObj, WordFile wordFile, HttpServletResponse response) throws UnsupportedEncodingException, FileNotFoundException {
|
// 数据map集合
|
Map<String, Object> wordData = new HashMap<>();
|
//HackLoopTableRenderPolicy hackLoopTableRenderPolicy = new HackLoopTableRenderPolicy();
|
Configure config = null;
|
ConfigureBuilder builder = Configure.newBuilder();
|
this.formatWordData(dataObj, wordData, builder);
|
|
config = builder.build();
|
File fl = new File(localPath + "/template/" + wordFile.getModulePath());
|
FileInputStream fs = new FileInputStream(fl);
|
XWPFTemplate template;
|
if (config != null) {
|
template = XWPFTemplate.compile(fs, config);
|
} else {
|
template = XWPFTemplate.compile(fs);
|
}
|
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);
|
}
|
}
|
}
|
