yhl452493373/document
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

初次提交

Browse Source
This commit is contained in:
yanghuanglin
2022-08-03 16:54:52 +08:00
parent 276c6f4a90
commit 22538415b9
19 changed files with 2511 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tl-server/src/main/java/com/optima/document/tl/server/DocumentServer.java Normal file
View File

@@ -0,0 +1,13 @@
package com.optima.document.tl.server;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
@SpringBootApplication
public class DocumentServer {
public static void main(String[] args) {
SpringApplication.run(DocumentServer.class, args);
}
}

170
tl-server/src/main/java/com/optima/document/tl/server/api/DocumentServiceImpl.java Normal file
View File

@@ -0,0 +1,170 @@
package com.optima.document.tl.server.api;
import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.config.Configure;
import com.deepoove.poi.data.PictureRenderData;
import com.deepoove.poi.data.TextRenderData;
import com.deepoove.poi.policy.ListRenderPolicy;
import com.deepoove.poi.policy.PictureRenderPolicy;
import com.deepoove.poi.render.RenderContext;
import com.deepoove.poi.template.run.RunTemplate;
import com.deepoove.poi.xwpf.BodyContainer;
import com.deepoove.poi.xwpf.BodyContainerFactory;
import com.optima.document.api.DocumentService;
import com.optima.document.tl.server.config.DocumentConfig;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.rendering.ImageType;
import org.apache.pdfbox.rendering.PDFRenderer;
import org.apache.pdfbox.tools.imageio.ImageIOUtil;
import org.apache.poi.xwpf.usermodel.XWPFRun;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.*;
/**
* 服务接口实现
* @author Elias
* @date 2021-09-28 16:18
*/
@Slf4j
@Component
public class DocumentServiceImpl implements DocumentService {
/**
* word模版引擎配置
*/
private static final Configure wtlConfig = Configure.builder().buildGramer("${", "}")
.setValidErrorHandler(new Configure.DiscardHandler())
.addPlugin('%', new ListRenderPolicy() {
@Override
public void doRender(RenderContext<List<Object>> context) throws Exception {
XWPFRun run = context.getRun();
List<?> dataList = context.getData();
Iterator<?> var5 = dataList.iterator();
while (var5.hasNext()) {
Object data = var5.next();
if (data instanceof TextRenderData) {
run.setText(((TextRenderData) data).getText());
if (var5.hasNext()) {
run.setText("，");
}
} else if (data instanceof PictureRenderData) {
PictureRenderPolicy.Helper.renderPicture(run, (PictureRenderData) data);
}
}
}
})
.setRenderDataComputeFactory(envModel ->
el -> {
Object data = envModel.getRoot();
if ("#this".equals(el)) {
return data;
} else if (data instanceof Map) {
Map dataMap = ((Map) data);
if (dataMap.containsKey(el)) {
return dataMap.get(el);
}
}
return null;
})
.build();
@Resource
private DocumentConfig documentConfig;
@Override
public byte[] generateWord(byte[] templateData, Map<String, Object> dataModel) {
long start = System.currentTimeMillis();
XWPFTemplate template = XWPFTemplate.compile(new ByteArrayInputStream(templateData), wtlConfig).render(dataModel);
try (ByteArrayOutputStream bos = new ByteArrayOutputStream()){
template.write(bos);
log.info("word generate========consuming：{} milliseconds", System.currentTimeMillis() - start);
return bos.toByteArray();
} catch (Exception e) {
log.error("word generate error", e);
return null;
}
}
@Override
public byte[] wordToPdf(byte[] source, boolean clear) {
try {
long start = System.currentTimeMillis();
if (clear) {
try (ByteArrayOutputStream bos = new ByteArrayOutputStream()) {
XWPFTemplate template = XWPFTemplate.compile(new ByteArrayInputStream(source), wtlConfig);
BodyContainer bodyContainer = BodyContainerFactory.getBodyContainer(template.getXWPFDocument());
template.getElementTemplates().forEach(it -> {
if (it instanceof RunTemplate) {
RunTemplate rt = (RunTemplate) it;
bodyContainer.clearPlaceholder(rt.getRun());
}
});
template.writeAndClose(bos);
source = bos.toByteArray();
log.info("清空占位符=======耗时：{} 毫秒", System.currentTimeMillis() - start);
} catch (Exception e) {
log.error("清空标签失败：", e);
return null;
}
}
Path tempFilePath = Files.createTempFile(UUID.randomUUID().toString(), ".docx");
Files.write(tempFilePath, source);
Path pdfPath = Files.createTempFile(UUID.randomUUID().toString(), ".pdf");
try {
long begin = System.currentTimeMillis();
String command = "%s -f %s -O %s -T wdFormatPDF";
command = String.format(command, documentConfig.getDocToProgram(), tempFilePath, pdfPath);
Process p = Runtime.getRuntime().exec(command);
p.waitFor();
long end = System.currentTimeMillis();
log.info("docto take time in millis:" + (end - begin));
return Files.readAllBytes(pdfPath);
} catch (Exception e) {
e.printStackTrace();
return null;
} finally {
Files.delete(tempFilePath);
Files.delete(pdfPath);
}
} catch (Exception e) {
log.error("word to pdf error", e);
return null;
}
}
@Override
public byte[] wordToImage(byte[] source, String targetFormat) {
try {
byte[] pdfBytes = wordToPdf(source, true);
long start = System.currentTimeMillis();
PDDocument document = PDDocument.load(new ByteArrayInputStream(pdfBytes));
PDFRenderer pdfRenderer = new PDFRenderer(document);
BufferedImage bim = pdfRenderer.renderImageWithDPI(
0, 300, ImageType.RGB);
ByteArrayOutputStream bos = new ByteArrayOutputStream();
ImageIOUtil.writeImage(bim, targetFormat, bos, 300);
document.close();
log.info("word to image=======consuming：{} milliseconds", System.currentTimeMillis() - start);
return bos.toByteArray();
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
}

41
tl-server/src/main/java/com/optima/document/tl/server/config/DocumentConfig.java Normal file
View File

@@ -0,0 +1,41 @@
package com.optima.document.tl.server.config;
import com.optima.document.api.DocumentService;
import com.optima.document.tl.server.api.DocumentServiceImpl;
import lombok.Data;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.remoting.httpinvoker.HttpInvokerServiceExporter;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;
/**
* 服务端配置
* @author Elias
* @date 2021-09-28 16:12
*/
@Configuration
@ConfigurationProperties(prefix = "document")
@Data
public class DocumentConfig {
private String docToProgram;
/**
* 文档接口
* @return httpinvoker
*/
@Bean(name = "/document-service")
HttpInvokerServiceExporter wordService(DocumentService documentService) {
HttpInvokerServiceExporter exporter = new HttpInvokerServiceExporter();
exporter.setService( documentService );
exporter.setServiceInterface(DocumentService.class);
return exporter;
}
}

4
tl-server/src/main/resources/application.yml Normal file
View File

@@ -0,0 +1,4 @@
server:
port: 9004
document:
doc-to-program: C:\\DocumentServer\\docto.exe

6
tl-server/src/main/resources/banner.txt Normal file
View File

@@ -0,0 +1,6 @@
██████╗ ██████╗ ██████╗██╗ ██╗███╗ ███╗███████╗███╗ ██╗████████╗
██╔══██╗██╔═══██╗██╔════╝██║ ██║████╗ ████║██╔════╝████╗ ██║╚══██╔══╝
██║ ██║██║ ██║██║ ██║ ██║██╔████╔██║█████╗ ██╔██╗ ██║ ██║
██║ ██║██║ ██║██║ ██║ ██║██║╚██╔╝██║██╔══╝ ██║╚██╗██║ ██║
██████╔╝╚██████╔╝╚██████╗╚██████╔╝██║ ╚═╝ ██║███████╗██║ ╚████║ ██║
╚═════╝ ╚═════╝ ╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚══════╝╚═╝ ╚═══╝ ╚═╝