123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113 |
- package com.yaoyicloud.render;
- import java.io.IOException;
- import java.io.InputStream;
- import java.time.LocalDate;
- import java.util.HashMap;
- import java.util.Map;
- import com.deepoove.poi.config.Configure;
- import com.deepoove.poi.config.ConfigureBuilder;
- import com.deepoove.poi.policy.RenderPolicy;
- import com.fasterxml.jackson.core.type.TypeReference;
- import com.fasterxml.jackson.databind.ObjectMapper;
- import com.google.protobuf.util.JsonFormat;
- import com.yaoyicloud.config.CommonDataCache;
- import com.yaoyicloud.config.FilerepoProperties;
- import com.yaoyicloud.message.FxyProtos;
- import lombok.extern.slf4j.Slf4j;
- /**
- * ServiceProviderInfo渲染器
- *
- */
- @Slf4j
- public final class ServiceProviderInfoRender extends AbstractRender {
- private final FilerepoProperties filerepoProperties;
- private final CommonDataCache commonDataCache;
- public ServiceProviderInfoRender(String cwd, FilerepoProperties filerepoProperties,
- CommonDataCache commonDataCache) {
- super(cwd);
- this.filerepoProperties = filerepoProperties;
- this.commonDataCache = commonDataCache;
- }
- @Override
- protected String getBasicPath() throws IOException {
- return filerepoProperties.getBasePath();
- }
- @Override
- protected String getReportImagePath() {
- return filerepoProperties.getReportImagePath();
- }
- /**
- * Docx 渲染
- *
- * @param info 数据
- * @param inputStream 模板内容
- * @return 本地文件目录
- * @throws IOException
- */
- public String renderDocx(String info, InputStream inputStream,
- String relationId, Integer counter) throws IOException {
- log.info("开始渲染公共封面模块,relationId: {}", relationId);
- // 配置POI-TL渲染器
- ConfigureBuilder builder = Configure.builder();
- RenderPolicy indicatorsRenderPolicy = this.getScoreRenderPolicy();
- builder.bind("checkItemScores", indicatorsRenderPolicy);
- FxyProtos.AuditResult.Builder auditResultBuilder = FxyProtos.AuditResult.newBuilder();
- JsonFormat.parser().merge(info, auditResultBuilder);
- FxyProtos.AuditResult defaultInstance = FxyProtos.AuditResult.getDefaultInstance();
- FxyProtos.AuditResult mergedProto = defaultInstance.toBuilder()
- .mergeFrom(auditResultBuilder.build())
- .build();
- String completeJson = JsonFormat.printer()
- .includingDefaultValueFields()
- .print(mergedProto);
- ObjectMapper objectMapper = new ObjectMapper();
- Map<String, Object> data = objectMapper.readValue(completeJson, new TypeReference<Map<String, Object>>() {});
- data.replaceAll((k, v) -> v.equals("") ? "-" : v);
- String tenantName = (String) data.get("tenantName");
- String name = (String) data.get("serviceProviderName");
- String type = (String) data.get("type");
- String reportDate = LocalDate.now().toString();
- HashMap<String, Object> commonData = new HashMap<>();
- commonData.put("tenantName", tenantName);
- commonData.put("type", type);
- commonData.put("reportDate", reportDate);
- commonData.put("name", name);
- commonDataCache.addDataMap(relationId, commonData);
- data.put("levelInteger", counter);
- data.putAll(commonData);
- fillDefaultValues(data);
- try {
- // 渲染文档
- String resultPath = this.renderDocx(data, inputStream, builder, relationId, "serviceProviderInfo");
- log.info("渲染公共封面成功,文件路径: {}", resultPath);
- return resultPath;
- } catch (Exception e) {
- log.error("渲染公共封面失败,relationId: {}", relationId, e);
- throw new IOException("文档渲染失败", e);
- }
- }
- /**
- * 填充默认值,确保所有必要字段都存在
- */
- private void fillDefaultValues(Map<String, Object> data) {
- Map<String, Object> basicInfoSummary = (Map<String, Object>) data.get("overallSummary");
- basicInfoSummary.replaceAll((k, v) -> v.equals("") ? "-" : v);
- }
- }
|