From 2f3d3fb97bd4ebdc00c40a2774465c8b3487b6d1 Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期三, 02 四月 2025 17:14:27 +0800
Subject: [PATCH] bug修改

---
 medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwMicroEquipmentRecordServiceImpl.java |  180 +++++++++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 153 insertions(+), 27 deletions(-)

diff --git a/medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwMicroEquipmentRecordServiceImpl.java b/medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwMicroEquipmentRecordServiceImpl.java
index 3c33577..735cdf0 100644
--- a/medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwMicroEquipmentRecordServiceImpl.java
+++ b/medicalWaste-system/src/main/java/com/sinata/system/service/impl/MwMicroEquipmentRecordServiceImpl.java
@@ -1,32 +1,41 @@
 package com.sinata.system.service.impl;
 
-import cn.idev.excel.EasyExcel;
+import cn.idev.excel.FastExcel;
 import com.alibaba.fastjson2.JSONObject;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.google.common.collect.Lists;
 import com.sinata.common.core.domain.entity.SysDictData;
 import com.sinata.common.entity.PageDTO;
+import com.sinata.common.utils.BeanUtils;
 import com.sinata.common.utils.CollUtils;
 import com.sinata.common.utils.DateUtils;
 import com.sinata.common.utils.SecurityUtils;
 import com.sinata.common.utils.StringUtils;
+import com.sinata.system.config.AutoColumnWidthStrategy;
 import com.sinata.system.domain.MwCollectRecord;
+import com.sinata.system.domain.MwMicroEquipment;
 import com.sinata.system.domain.MwMicroEquipmentRecord;
 import com.sinata.system.domain.MwMicroEquipmentRecordItem;
+import com.sinata.system.domain.SysDepartment;
 import com.sinata.system.domain.dto.MwMicroEquipmentRecordDTO;
 import com.sinata.system.domain.query.MwMicroEquipmentRecordQuery;
 import com.sinata.system.domain.query.MwMicroEquipmentStaticsQuery;
+import com.sinata.system.domain.vo.DepartmentReportItemVO;
+import com.sinata.system.domain.vo.DepartmentReportVO;
 import com.sinata.system.domain.vo.MwCollectRecordVO;
 import com.sinata.system.domain.vo.MwMicroEquipmentRecordVO;
 import com.sinata.system.domain.vo.MwMicroEquipmentStaticsTitleVO;
 import com.sinata.system.domain.vo.MwMicroEquipmentStaticsVO;
+import com.sinata.system.domain.vo.SysDictDataVO;
 import com.sinata.system.enums.MedicalWasteStatusEnum;
 import com.sinata.system.mapper.MwMicroEquipmentRecordMapper;
 import com.sinata.system.service.ISysDictDataService;
 import com.sinata.system.service.MwCollectRecordService;
 import com.sinata.system.service.MwMicroEquipmentRecordItemService;
 import com.sinata.system.service.MwMicroEquipmentRecordService;
+import com.sinata.system.service.MwMicroEquipmentService;
+import com.sinata.system.service.MwStagingRoomService;
 import com.sinata.system.service.SysDepartmentService;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -41,6 +50,8 @@
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
+import java.util.Objects;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -57,6 +68,8 @@
     private final MwCollectRecordService mwCollectRecordService;
     private final MwMicroEquipmentRecordItemService mwMicroEquipmentRecordItemService;
     private final ISysDictDataService sysDictDataService;
+    private final MwStagingRoomService mwStagingRoomService;
+    private final MwMicroEquipmentService mwMicroEquipmentService;
 
     /**
      * 设备使用记录分页列表
@@ -71,6 +84,20 @@
             return PageDTO.empty(0L, 0L);
         }
         return PageDTO.of(baseMapper.pageList(new Page<>(query.getPageCurr(), query.getPageSize()), query, treeCode));
+    }
+
+    @Override
+    public MwMicroEquipmentRecordVO detail(Long id) {
+        MwMicroEquipmentRecordVO mwMicroEquipmentRecordVO = BeanUtils.copyBean(this.getById(id), MwMicroEquipmentRecordVO.class);
+        if (Objects.nonNull(mwMicroEquipmentRecordVO)) {
+            MwMicroEquipment equipment = mwMicroEquipmentService.getById(
+                    mwMicroEquipmentRecordVO.getEquipmentId());
+            if (Objects.nonNull(equipment)) {
+                mwMicroEquipmentRecordVO.setEquipmentName(equipment.getEquipmentName());
+            }
+            mwMicroEquipmentRecordVO.setMedicalWasteList(mwStagingRoomService.queryMedicalWasteList(mwMicroEquipmentRecordVO.getId()));
+        }
+        return mwMicroEquipmentRecordVO;
     }
 
     /**
@@ -92,7 +119,9 @@
         // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
         String fileName = URLEncoder.encode("微波设备使用记录", "UTF-8").replaceAll("\\+", "%20");
         response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
-        EasyExcel.write(response.getOutputStream(), MwCollectRecordVO.class).sheet("微波设备使用记录").doWrite(list);
+        FastExcel.write(response.getOutputStream(), MwCollectRecordVO.class)
+                .registerWriteHandler(new AutoColumnWidthStrategy())
+                .sheet("微波设备使用记录").doWrite(list);
     }
 
     /**
@@ -172,43 +201,112 @@
      * @return
      */
     @Override
-    public List<List<String>> getStaticsData(MwMicroEquipmentStaticsQuery query) {
+    public DepartmentReportVO getStaticsData(MwMicroEquipmentStaticsQuery query) {
+        DepartmentReportVO vo = new DepartmentReportVO();
+        SysDepartment department = sysDepartmentService.getById(query.getDepartmentId());
+        if (Objects.nonNull(department)) {
+            vo.setDepartmentName(department.getDepartmentName());
+        }
+        MwMicroEquipment microEquipment = mwMicroEquipmentService.getById(query.getEquipmentId());
+        vo.setEquipmentName(microEquipment.getEquipmentName());
+        List<SysDictDataVO> wasteTypeList = sysDictDataService.medicalWasteTypeList();
+
+        if (CollUtils.isEmpty(wasteTypeList)) {
+            return vo;
+        }
+        vo.setLegend(wasteTypeList.stream().map(SysDictDataVO::getDictLabel).collect(Collectors.toList()));
+        vo.getLegend().add("小计");
         String treeCode = sysDepartmentService.getTreeCodeByDepartmentId(query.getDepartmentId());
-        List<SysDictData> medicalWasteTypeList = sysDictDataService.lambdaQuery().eq(SysDictData::getDictType, "medical_waste_type").list();
         List<MwMicroEquipmentStaticsVO> staticsData = baseMapper.getStaticsData(query, treeCode);
         SimpleDateFormat sdf = new SimpleDateFormat(DateUtils.YYYY_MM_DD_HH_MM_SS);
         switch (query.getDateType()) {
             case 1:
                 sdf = new SimpleDateFormat(DateUtils.YYYY_MM_DD);
+                break;
             case 2:
                 sdf = new SimpleDateFormat(DateUtils.YYYY_MM);
+                break;
             case 3:
                 sdf = new SimpleDateFormat(DateUtils.YYYY);
+                break;
         }
         List<String> dateList = DateUtils.getDayBetween(query.getStartTime(), query.getEndTime(), query.getDateType());
-        List<List<String>> result = new ArrayList<>();
-        if (CollUtils.isNotEmpty(staticsData)) {
-            for (String date : dateList) {
-                List<String> list = new ArrayList<>();
-                list.add(date);
-                SimpleDateFormat finalSdf = sdf;
-                for (SysDictData sysDictData : medicalWasteTypeList) {
-                    BigDecimal weight = staticsData.stream().filter(e -> e.getWasteType().equals(sysDictData.getDictCode())
-                            && finalSdf.format(e.getUseTime()).equals(date)
-                    ).map(MwMicroEquipmentStaticsVO::getWeight).reduce(BigDecimal::add).orElse(BigDecimal.ZERO).setScale(2, RoundingMode.HALF_UP);
-                    long count = staticsData.stream().filter(e -> e.getWasteType().equals(sysDictData.getDictCode())
-                            && finalSdf.format(e.getUseTime()).equals(date)).count();
-                    list.add(weight.toString());
-                    list.add(String.valueOf(count));
-                }
-                staticsData.stream().filter(e -> finalSdf.format(e.getUseTime()).equals(date)).findFirst().ifPresent(item -> {
-                    list.add(item.getTotalWeight().toString());
-                    list.add(String.valueOf(item.getBagNum()));
-                });
-                result.add(list);
+        List<DepartmentReportItemVO> result = new ArrayList<>();
+        for (String date : dateList) {
+            DepartmentReportItemVO departmentReportItemVO = new DepartmentReportItemVO();
+            departmentReportItemVO.setName(date);
+            departmentReportItemVO.setData(new ArrayList<>());
+            SimpleDateFormat finalSdf = sdf;
+            BigDecimal totalWeight = BigDecimal.ZERO;
+            BigDecimal totalCount = BigDecimal.ZERO;
+            for (SysDictDataVO sysDictData : wasteTypeList) {
+                BigDecimal weight = staticsData.stream().filter(e -> e.getWasteType().equals(sysDictData.getDictCode())
+                        && finalSdf.format(e.getUseTime()).equals(date)
+                ).map(MwMicroEquipmentStaticsVO::getWeight).reduce(BigDecimal.ZERO, BigDecimal::add).setScale(2, RoundingMode.HALF_UP);
+                long count = staticsData.stream().filter(e -> e.getWasteType().equals(sysDictData.getDictCode())
+                        && finalSdf.format(e.getUseTime()).equals(date)).count();
+                departmentReportItemVO.getData().add(weight);
+                departmentReportItemVO.getData().add(BigDecimal.valueOf(count));
+                totalWeight = totalWeight.add(weight);
+                totalCount = totalCount.add(BigDecimal.valueOf(count));
             }
+            departmentReportItemVO.getData().add(totalWeight);
+            departmentReportItemVO.getData().add(totalCount);
+            result.add(departmentReportItemVO);
         }
-        return result;
+
+        // 添加合计行
+        if (!result.isEmpty()) {
+            DepartmentReportItemVO totalRow = new DepartmentReportItemVO();
+            totalRow.setName("合计");
+            totalRow.setData(new ArrayList<>());
+
+            // 计算各类型废物的重量和数量总计
+            for (SysDictDataVO sysDictData : wasteTypeList) {
+                BigDecimal totalTypeWeight = BigDecimal.ZERO;
+                BigDecimal totalTypeCount = BigDecimal.ZERO;
+
+                for (DepartmentReportItemVO item : result) {
+                    // 每种废物类型占用两列(重量和数量)
+                    int index = wasteTypeList.indexOf(sysDictData);
+                    int weightIndex = index * 2;
+                    int countIndex = weightIndex + 1;
+
+                    if (weightIndex < item.getData().size()) {
+                        totalTypeWeight = totalTypeWeight.add(item.getData().get(weightIndex));
+                    }
+                    if (countIndex < item.getData().size()) {
+                        totalTypeCount = totalTypeCount.add(item.getData().get(countIndex));
+                    }
+                }
+
+                totalRow.getData().add(totalTypeWeight);
+                totalRow.getData().add(totalTypeCount);
+            }
+
+            // 计算总的重量和数量
+            BigDecimal finalTotalWeight = BigDecimal.ZERO;
+            BigDecimal finalTotalCount = BigDecimal.ZERO;
+
+            for (DepartmentReportItemVO item : result) {
+                int totalWeightIndex = item.getData().size() - 2;
+                int totalCountIndex = item.getData().size() - 1;
+
+                if (totalWeightIndex >= 0 && totalCountIndex >= 0) {
+                    finalTotalWeight = finalTotalWeight.add(item.getData().get(totalWeightIndex));
+                    finalTotalCount = finalTotalCount.add(item.getData().get(totalCountIndex));
+                }
+            }
+
+            totalRow.getData().add(finalTotalWeight);
+            totalRow.getData().add(finalTotalCount);
+
+            // 将合计行添加到列表开头
+            result.add(0, totalRow);
+        }
+
+        vo.setList(result);
+        return vo;
     }
 
     @Override
@@ -223,13 +321,40 @@
         String fileName = URLEncoder.encode("处置分析报表", "UTF-8").replaceAll("\\+", "%20");
         response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
         // 这里需要设置不关闭流
-        EasyExcel.write(response.getOutputStream())
+        FastExcel.write(response.getOutputStream())
                 .head(head)
                 .autoCloseStream(Boolean.TRUE)
+                .registerWriteHandler(new AutoColumnWidthStrategy())
                 .sheet("处置分析报表")
-                .doWrite(getStaticsData(query));
+                .doWrite(getStaticsReportData(query));
     }
 
+    /**
+     * 导出报表数据
+     *
+     * @param query
+     * @return
+     */
+    private List<List<Object>> getStaticsReportData(MwMicroEquipmentStaticsQuery query) {
+        DepartmentReportVO vo = getStaticsData(query);
+        if (CollUtils.isNotEmpty(vo.getList())) {
+            List<List<Object>> result = new ArrayList<>();
+            for (DepartmentReportItemVO departmentReportItemVO : vo.getList()) {
+                List<Object> data = new ArrayList<>();
+                data.add(departmentReportItemVO.getName());
+                data.addAll(departmentReportItemVO.getData());
+                result.add(data);
+            }
+            return result;
+        }
+        return CollUtils.emptyList();
+    }
+
+    /**
+     * 导出报表表头
+     *
+     * @return
+     */
     private List<List<String>> head() {
         List<SysDictData> medicalWasteTypeList = sysDictDataService.lambdaQuery().eq(SysDictData::getDictType, "medical_waste_type").list();
         List<List<String>> headTitles = Lists.newArrayList();
@@ -242,4 +367,5 @@
         headTitles.add(Lists.newArrayList("小计", "总数量"));
         return headTitles;
     }
+
 }

--
Gitblit v1.7.1