监管报告

src/main/java/com/izouma/wenlvju/dto/ReportDTO.java

@@ -1,8 +1,32 @@
 package com.izouma.wenlvju.dto;
 
+import com.izouma.wenlvju.enums.UnitType;
 import io.swagger.annotations.ApiModelProperty;
+import lombok.Builder;
+import lombok.Data;
 
+import javax.persistence.EnumType;
+import javax.persistence.Enumerated;
+
+@Data
+@Builder
 public class ReportDTO {
+    @Enumerated(EnumType.STRING)
+    @ApiModelProperty(value = "单位类型")
+    private UnitType unitType;
+
+    @ApiModelProperty(value = "名称")
+    private String name;
+
     @ApiModelProperty(value = "投诉条数")
     private int complain;
+
+    @ApiModelProperty(value = "投诉内容")
+    private String complainContent;
+
+    @ApiModelProperty(value = "考级数量")
+    private int examCenterQuantity;
+
+    @ApiModelProperty(value = "报考人数")
+    private int examQuantity;
 }

src/main/java/com/izouma/wenlvju/repo/RecordRepo.java

@@ -18,7 +18,7 @@ public interface RecordRepo extends JpaRepository<Record, Long>, JpaSpecificatio
 
     Record findFirstByDelIsTrue();
 
-    List<Record> findAllByExaminationStartTimeBetween(LocalDate startTime, LocalDate endTime);
+    List<Record> findAllByExaminationStartTimeBetween(LocalDate start, LocalDate end);
 
 //    List<Record> findAllBySupervisorUserId(Long supervisorUserId);
 

src/main/java/com/izouma/wenlvju/repo/regulation/ComplainRepo.java

@@ -7,10 +7,14 @@ import org.springframework.data.jpa.repository.Modifying;
 import org.springframework.data.jpa.repository.Query;
 
 import javax.transaction.Transactional;
+import java.time.LocalDateTime;
+import java.util.List;
 
 public interface ComplainRepo extends JpaRepository<Complain, Long>, JpaSpecificationExecutor<Complain> {
     @Query("update Complain t set t.del = true where t.id = ?1")
     @Modifying
     @Transactional
     void softDelete(Long id);
+
+    List<Complain> findAllByComplainAtBetween(LocalDateTime start, LocalDateTime end);
 }

src/main/java/com/izouma/wenlvju/service/regulation/ReportService.java

@@ -1,16 +1,22 @@
 package com.izouma.wenlvju.service.regulation;
 
+import com.izouma.wenlvju.domain.GradingOrganization;
 import com.izouma.wenlvju.domain.Record;
+import com.izouma.wenlvju.domain.regulation.Complain;
 import com.izouma.wenlvju.domain.regulation.Report;
 import com.izouma.wenlvju.dto.PageQuery;
+import com.izouma.wenlvju.dto.ReportDTO;
+import com.izouma.wenlvju.enums.UnitType;
+import com.izouma.wenlvju.repo.GradingOrganizationRepo;
 import com.izouma.wenlvju.repo.RecordRepo;
+import com.izouma.wenlvju.repo.regulation.ComplainRepo;
 import com.izouma.wenlvju.repo.regulation.ReportRepo;
 import com.izouma.wenlvju.utils.JpaUtils;
 import lombok.AllArgsConstructor;
 import org.springframework.data.domain.Page;
 import org.springframework.stereotype.Service;
 
-import java.time.LocalDate;
+import java.time.LocalDateTime;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
@@ -20,8 +26,10 @@ import java.util.stream.Collectors;
 @AllArgsConstructor
 public class ReportService {
 
-    private ReportRepo reportRepo;
-    private RecordRepo recordRepo;
+    private ReportRepo              reportRepo;
+    private RecordRepo              recordRepo;
+    private ComplainRepo            complainRepo;
+    private GradingOrganizationRepo gradingOrganizationRepo;
 
     public Page<Report> all(PageQuery pageQuery) {
         return reportRepo.findAll(JpaUtils.toSpecification(pageQuery, Report.class), JpaUtils.toPageRequest(pageQuery));
@@ -32,20 +40,51 @@ public class ReportService {
      */
     public void generate(Report report) {
         List<Record> records = new ArrayList<>();
+        List<Complain> complains = new ArrayList<>();
         switch (report.getType()) {
             case ANNUAL:
                 int year = Integer.parseInt(report.getYear());
-                LocalDate start = LocalDate.of(year, 1, 1);
-                LocalDate end = LocalDate.of(year, 12, 31);
-                recordRepo.findAllByExaminationStartTimeBetween(start, end);
-
+                LocalDateTime start = LocalDateTime.of(year, 1, 1, 0, 0);
+                LocalDateTime end = LocalDateTime.of(year, 12, 31, 23, 59, 59, 59);
+                records = recordRepo.findAllByExaminationStartTimeBetween(start.toLocalDate(), end.toLocalDate());
+                complains = complainRepo.findAllByComplainAtBetween(start, end);
                 break;
             case QUARTERLY:
                 break;
             case MONTHLY:
                 break;
         }
+        Map<String, GradingOrganization> gradingOrganizationMap = gradingOrganizationRepo.findAll()
+                .stream()
+                .collect(Collectors.toMap(GradingOrganization::getName, go -> go));
+        Map<String, List<Record>> examinationAgency = records.stream()
+                .collect(Collectors.groupingBy(Record::getExaminationAgency));
+
+        Map<String, List<Record>> organizer = records.stream()
+                .collect(Collectors.groupingBy(Record::getOrganizer));
+
+        Map<Long, List<Complain>> complainsByName = complains.stream()
+                .collect(Collectors.groupingBy(Complain::getUnitName));
+
+        List<ReportDTO> reportDTOS = new ArrayList<>();
+        examinationAgency.forEach((key, value) -> {
+            GradingOrganization go = gradingOrganizationMap.get(key);
+            // 投诉
+            List<Complain> complainList = complainsByName.get(go.getId());
+            String complainContent = complainList.stream().map(Complain::getTitle).collect(Collectors.joining(","));
+            // 备案
+            List<Record> recordList = examinationAgency.get(key);
+            int examCenterQuantity = recordList.stream().mapToInt(Record::getExamCenterQuantity).sum();
+            int examQuantity = recordList.stream().mapToInt(Record::getExamQuantity).sum();
 
-//        records.forEach();
+            reportDTOS.add(ReportDTO.builder()
+                    .unitType(UnitType.GRADING_ORGANIZATION)
+                    .name(key)
+                    .complain(complainList.size())
+                    .complainContent(complainContent)
+                    .examCenterQuantity(examCenterQuantity)
+                    .examQuantity(examQuantity)
+                    .build());
+        });
     }
 }