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;
 
 /**
@@ -636,11 +639,14 @@ 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());
@@ -654,9 +660,11 @@ public class EvaluationServiceImpl implements EvaluationService {
         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);
 
+        // 如果分布为空,则填充默认值 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()));
@@ -664,6 +672,10 @@ public class EvaluationServiceImpl implements EvaluationService {
         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));
+    }
+}
+