refactor: 重构 PDF 模板

src/main/java/cn/teammodel/service/impl/EvaluationServiceImpl.java

@@ -31,6 +31,7 @@ import org.apache.commons.lang3.ObjectUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.jfree.chart.ChartUtils;
 import org.jfree.chart.JFreeChart;
+import org.jfree.chart.plot.PiePlot;
 import org.jfree.data.general.DefaultPieDataset;
 import org.springframework.core.io.ClassPathResource;
 import org.springframework.data.domain.Page;
@@ -40,12 +41,14 @@ import org.springframework.stereotype.Service;
 import javax.annotation.Resource;
 import javax.servlet.ServletOutputStream;
 import javax.servlet.http.HttpServletResponse;
+import java.awt.*;
 import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.time.Instant;
 import java.time.LocalDate;
 import java.util.*;
+import java.util.List;
 import java.util.stream.Collectors;
 
 /**
@@ -332,7 +335,7 @@ public class EvaluationServiceImpl implements EvaluationService {
             // 获取班级信息
             ClassInfo classInfo = classRepository.findClassByIdAndCode(classId, String.format(PK.CLASS, schoolId));
             className = classInfo.getName();
-        } else if (targetType.equals(TARGET_CLASS)){
+        } else if (targetType.equals(TARGET_CLASS)) {
             ClassInfo classInfo = classRepository.findClassByIdAndCode(targetId, String.format(PK.CLASS, schoolId));
             if (classInfo == null) {
                 throw new ServiceException(ErrorCode.PARAMS_ERROR.getCode(), "班级不存在");
@@ -365,7 +368,7 @@ public class EvaluationServiceImpl implements EvaluationService {
                 targetId,
                 classId,
                 academicYearId,
-                String.format(PK.PK_APPRAISE_RECORD,schoolId)
+                String.format(PK.PK_APPRAISE_RECORD, schoolId)
         );
 
         // 初始化新的评价节点
@@ -376,7 +379,7 @@ public class EvaluationServiceImpl implements EvaluationService {
         item.setCreatorId(loginUser.getId());
         item.setCreateTime(Instant.now().toEpochMilli());
         // 处理学校与学生的差异
-        if (targetType.equals(TARGET_CLASS)){
+        if (targetType.equals(TARGET_CLASS)) {
             item.setSpread(spread);
         } else {
             item.setPushParent(pushParent);
@@ -530,7 +533,7 @@ public class EvaluationServiceImpl implements EvaluationService {
             AppraiseTreeNode appraiseNode = record.getAppraiseNode();
             String[] path = appraiseNode.getPath();
             // 数据异常则跳过
-            if (path == null || path.length == 0){
+            if (path == null || path.length == 0) {
                 continue;
             }
             String root = path[0];
@@ -620,7 +623,7 @@ public class EvaluationServiceImpl implements EvaluationService {
     }
 
     @Override
-    public void exportStuReportPdf(IdRequest idRequest, HttpServletResponse response) throws IOException, DocumentException{
+    public void exportStuReportPdf(IdRequest idRequest, HttpServletResponse response) throws IOException, DocumentException {
         // 设置response参数
         response.reset();
         response.setContentType("application/pdf");
@@ -636,34 +639,43 @@ public class EvaluationServiceImpl implements EvaluationService {
         PdfReader pdfReader = new PdfReader(in);
         PdfStamper stamper = new PdfStamper(pdfReader, os);
 
+        int criticalCount = reportVo.getCriticalDistribution().values().stream().mapToInt(Integer::intValue).sum();
         Map<String, String> data = new HashMap<>();
         data.put("name", reportVo.getName());
         data.put("studentId", reportVo.getStudentId());
         data.put("className", reportVo.getClassName());
+        data.put("totalCount", String.valueOf(reportVo.getPraiseCount() + criticalCount));
         data.put("praiseCount", String.valueOf(reportVo.getPraiseCount()));
+        data.put("criticalCount", String.valueOf(criticalCount));
         data.put("beyondPercent", String.valueOf(reportVo.getBeyondPercent()));
         data.put("topPraiseTeacher", StringUtils.isBlank(reportVo.getTopPraiseTeacher().getName()) ? "暂无" : reportVo.getTopPraiseTeacher().getName());
         data.put("topCriticalTeacher", StringUtils.isBlank(reportVo.getTopCriticalTeacher().getName()) ? "暂无" : reportVo.getTopCriticalTeacher().getName());
         data.put("topPraiseNode", StringUtils.isBlank(reportVo.getTopPraiseNode()) ? "暂无" : reportVo.getTopPraiseNode());
         data.put("topCriticalNode", StringUtils.isBlank(reportVo.getTopCriticalNode()) ? "暂无" : reportVo.getTopCriticalNode());
 
-        DefaultPieDataset praiseDistributionDataset = new DefaultPieDataset( );
+        DefaultPieDataset praiseDistributionDataset = new DefaultPieDataset();
         praiseDistributionDataset.setValue(FiveEducations.VIRTUE.getName(), reportVo.getPraiseDistribution().get(FiveEducations.VIRTUE.getCode()));
         praiseDistributionDataset.setValue(FiveEducations.INTELLIGENCE.getName(), reportVo.getPraiseDistribution().get(FiveEducations.INTELLIGENCE.getCode()));
         praiseDistributionDataset.setValue(FiveEducations.SPORTS.getName(), reportVo.getPraiseDistribution().get(FiveEducations.SPORTS.getCode()));
         praiseDistributionDataset.setValue(FiveEducations.ART.getName(), reportVo.getPraiseDistribution().get(FiveEducations.ART.getCode()));
         praiseDistributionDataset.setValue(FiveEducations.LABOUR.getName(), reportVo.getPraiseDistribution().get(FiveEducations.LABOUR.getCode()));
         JFreeChart praisePieChart = ChartUtil.pieChart("五育表扬指标分布", praiseDistributionDataset);
+
         ByteArrayOutputStream praiseBos = new ByteArrayOutputStream();
         ChartUtils.writeChartAsJPEG(praiseBos, praisePieChart, 340, 330);
 
-        DefaultPieDataset criticalDistributionDataset = new DefaultPieDataset( );
+        // 如果分布为空,则填充默认值 1
+        DefaultPieDataset criticalDistributionDataset = new DefaultPieDataset();
         criticalDistributionDataset.setValue(FiveEducations.VIRTUE.getName(), reportVo.getCriticalDistribution().get(FiveEducations.VIRTUE.getCode()));
         criticalDistributionDataset.setValue(FiveEducations.INTELLIGENCE.getName(), reportVo.getCriticalDistribution().get(FiveEducations.INTELLIGENCE.getCode()));
         criticalDistributionDataset.setValue(FiveEducations.SPORTS.getName(), reportVo.getCriticalDistribution().get(FiveEducations.SPORTS.getCode()));
         criticalDistributionDataset.setValue(FiveEducations.ART.getName(), reportVo.getCriticalDistribution().get(FiveEducations.ART.getCode()));
         criticalDistributionDataset.setValue(FiveEducations.LABOUR.getName(), reportVo.getCriticalDistribution().get(FiveEducations.LABOUR.getCode()));
         JFreeChart criticalPieChart = ChartUtil.pieChart("五育待改进指标分布", criticalDistributionDataset);
+        PiePlot plot = (PiePlot) criticalPieChart.getPlot();
+        plot.setNoDataMessage("无数据");
+        plot.setNoDataMessageFont(new Font("宋体", Font.PLAIN, 18));
+
         ByteArrayOutputStream criticalBos = new ByteArrayOutputStream();
         ChartUtils.writeChartAsJPEG(criticalBos, criticalPieChart, 340, 330);
 
@@ -675,7 +687,6 @@ public class EvaluationServiceImpl implements EvaluationService {
 //        stamper.setFormFlattening(true);
         stamper.close();
         os.close();
-
     }
 
     /**
@@ -723,4 +734,24 @@ public class EvaluationServiceImpl implements EvaluationService {
         }
         return children;
     }
+
+    /**
+    *   保证饼图分布：distribution 全部为 0,则初始化为 1
+    */
+    private void checkIfDistributionNull(Map<String, Integer> distribution) {
+        boolean flag = false;
+        for (Map.Entry<String, Integer> entry : distribution.entrySet()) {
+            if (entry.getValue() != 0) {
+                flag = true;
+                break;
+            }
+        }
+
+        if(flag) {
+            return;
+        }
+        // distribution 全部为 0,则初始化为 1
+        distribution.forEach((k, v) -> distribution.put(k, 1));
+    }
 }
+