ServiceProviderInfoRender.java 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113
  1. package com.yaoyicloud.render;
  2. import java.io.IOException;
  3. import java.io.InputStream;
  4. import java.time.LocalDate;
  5. import java.util.HashMap;
  6. import java.util.Map;
  7. import com.deepoove.poi.config.Configure;
  8. import com.deepoove.poi.config.ConfigureBuilder;
  9. import com.deepoove.poi.policy.RenderPolicy;
  10. import com.fasterxml.jackson.core.type.TypeReference;
  11. import com.fasterxml.jackson.databind.ObjectMapper;
  12. import com.google.protobuf.util.JsonFormat;
  13. import com.yaoyicloud.config.CommonDataCache;
  14. import com.yaoyicloud.config.FilerepoProperties;
  15. import com.yaoyicloud.message.FxyProtos;
  16. import lombok.extern.slf4j.Slf4j;
  17. /**
  18. * ServiceProviderInfo渲染器
  19. *
  20. */
  21. @Slf4j
  22. public final class ServiceProviderInfoRender extends AbstractRender {
  23. private final FilerepoProperties filerepoProperties;
  24. private final CommonDataCache commonDataCache;
  25. public ServiceProviderInfoRender(String cwd, FilerepoProperties filerepoProperties,
  26. CommonDataCache commonDataCache) {
  27. super(cwd);
  28. this.filerepoProperties = filerepoProperties;
  29. this.commonDataCache = commonDataCache;
  30. }
  31. @Override
  32. protected String getBasicPath() throws IOException {
  33. return filerepoProperties.getBasePath();
  34. }
  35. @Override
  36. protected String getReportImagePath() {
  37. return filerepoProperties.getReportImagePath();
  38. }
  39. /**
  40. * Docx 渲染
  41. *
  42. * @param info 数据
  43. * @param inputStream 模板内容
  44. * @return 本地文件目录
  45. * @throws IOException
  46. */
  47. public String renderDocx(String info, InputStream inputStream,
  48. String relationId, Integer counter) throws IOException {
  49. log.info("开始渲染公共封面模块,relationId: {}", relationId);
  50. // 配置POI-TL渲染器
  51. ConfigureBuilder builder = Configure.builder();
  52. RenderPolicy indicatorsRenderPolicy = this.getScoreRenderPolicy();
  53. builder.bind("checkItemScores", indicatorsRenderPolicy);
  54. FxyProtos.AuditResult.Builder auditResultBuilder = FxyProtos.AuditResult.newBuilder();
  55. JsonFormat.parser().merge(info, auditResultBuilder);
  56. FxyProtos.AuditResult defaultInstance = FxyProtos.AuditResult.getDefaultInstance();
  57. FxyProtos.AuditResult mergedProto = defaultInstance.toBuilder()
  58. .mergeFrom(auditResultBuilder.build())
  59. .build();
  60. String completeJson = JsonFormat.printer()
  61. .includingDefaultValueFields()
  62. .print(mergedProto);
  63. ObjectMapper objectMapper = new ObjectMapper();
  64. Map<String, Object> data = objectMapper.readValue(completeJson, new TypeReference<Map<String, Object>>() {});
  65. data.replaceAll((k, v) -> v.equals("") ? "-" : v);
  66. String tenantName = (String) data.get("tenantName");
  67. String name = (String) data.get("serviceProviderName");
  68. String type = (String) data.get("type");
  69. String reportDate = LocalDate.now().toString();
  70. HashMap<String, Object> commonData = new HashMap<>();
  71. commonData.put("tenantName", tenantName);
  72. commonData.put("type", type);
  73. commonData.put("reportDate", reportDate);
  74. commonData.put("name", name);
  75. commonDataCache.addDataMap(relationId, commonData);
  76. data.put("levelInteger", counter);
  77. data.putAll(commonData);
  78. fillDefaultValues(data);
  79. try {
  80. // 渲染文档
  81. String resultPath = this.renderDocx(data, inputStream, builder, relationId, "serviceProviderInfo");
  82. log.info("渲染公共封面成功,文件路径: {}", resultPath);
  83. return resultPath;
  84. } catch (Exception e) {
  85. log.error("渲染公共封面失败,relationId: {}", relationId, e);
  86. throw new IOException("文档渲染失败", e);
  87. }
  88. }
  89. /**
  90. * 填充默认值,确保所有必要字段都存在
  91. */
  92. private void fillDefaultValues(Map<String, Object> data) {
  93. Map<String, Object> basicInfoSummary = (Map<String, Object>) data.get("overallSummary");
  94. basicInfoSummary.replaceAll((k, v) -> v.equals("") ? "-" : v);
  95. }
  96. }