From a26151eff700b514ca92b0ac9207200b866071fc Mon Sep 17 00:00:00 2001
From: zss <zss@example.com>
Date: 星期四, 30 五月 2024 15:24:48 +0800
Subject: [PATCH] 工时统计+权限+异步处理method+检验样品结果判断

---
 performance-server/src/main/java/com/yuanchu/mom/service/impl/AuxiliaryOutputWorkingHoursServiceImpl.java |  185 +++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 182 insertions(+), 3 deletions(-)

diff --git a/performance-server/src/main/java/com/yuanchu/mom/service/impl/AuxiliaryOutputWorkingHoursServiceImpl.java b/performance-server/src/main/java/com/yuanchu/mom/service/impl/AuxiliaryOutputWorkingHoursServiceImpl.java
index 07f0be7..891c9ef 100644
--- a/performance-server/src/main/java/com/yuanchu/mom/service/impl/AuxiliaryOutputWorkingHoursServiceImpl.java
+++ b/performance-server/src/main/java/com/yuanchu/mom/service/impl/AuxiliaryOutputWorkingHoursServiceImpl.java
@@ -1,18 +1,37 @@
 package com.yuanchu.mom.service.impl;
 
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
+import com.alibaba.excel.write.metadata.WriteSheet;
+import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
+import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.yuanchu.mom.common.GetLook;
 import com.yuanchu.mom.common.PrintChina;
+import com.yuanchu.mom.dto.AuxiliaryOutputWorkingHoursDto;
+import com.yuanchu.mom.dto.AuxiliaryWorkingHoursDayDto;
+import com.yuanchu.mom.mapper.AuxiliaryWorkingHoursDayMapper;
+import com.yuanchu.mom.mapper.PowerMapper;
+import com.yuanchu.mom.mapper.UserMapper;
 import com.yuanchu.mom.pojo.AuxiliaryOutputWorkingHours;
 import com.yuanchu.mom.mapper.AuxiliaryOutputWorkingHoursMapper;
 import com.yuanchu.mom.pojo.AuxiliaryWorkingHoursDay;
+import com.yuanchu.mom.pojo.Power;
+import com.yuanchu.mom.pojo.User;
 import com.yuanchu.mom.service.AuxiliaryOutputWorkingHoursService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yuanchu.mom.utils.QueryWrappers;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.net.URLEncoder;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -28,11 +47,171 @@
     @Resource
     AuxiliaryOutputWorkingHoursMapper auxiliaryOutputWorkingHoursMapper;
 
+    @Resource
+    GetLook getLook;
+
+    @Resource
+    AuxiliaryWorkingHoursDayMapper auxiliaryWorkingHoursDayMapper;
+
+    @Resource
+    UserMapper userMapper;
+
+    @Resource
+    PowerMapper powerMapper;
+
     @Override
-    public Map<String, Object> selectAuxiliaryOutputWorkingHours(Page page, AuxiliaryOutputWorkingHours auxiliaryOutputWorkingHours) {
+    public Map<String, Object> selectAuxiliaryOutputWorkingHours(Page page, AuxiliaryOutputWorkingHoursDto auxiliaryOutputWorkingHoursDto) {
+        String dates = auxiliaryOutputWorkingHoursDto.getDateTime();
+        String week = auxiliaryOutputWorkingHoursDto.getWeek();
+        auxiliaryOutputWorkingHoursDto.setDateTime(null);
+        auxiliaryOutputWorkingHoursDto.setWeek(null);
         Map<String, Object> map = new HashMap<>();
-        map.put("head", PrintChina.printChina(AuxiliaryOutputWorkingHours.class));
-        map.put("body", auxiliaryOutputWorkingHoursMapper.selectAuxiliaryOutputWorkingHours(page, QueryWrappers.queryWrappers(auxiliaryOutputWorkingHours)));
+        List<Integer> ids = null;
+        map.put("head", PrintChina.printChina(AuxiliaryOutputWorkingHoursDto.class));
+        //鍒ゆ柇缁勯暱,缁勫憳,绠$悊鍛樻潈闄�
+        Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectAuxiliaryOutputWorkingHours");
+        User user = userMapper.selectById(map1.get("userId"));
+        Integer roleId = user.getRoleId();
+        Power power = powerMapper.selectOne(Wrappers.<Power>lambdaQuery().eq(Power::getRoleId, roleId).eq(Power::getMenuMethod, "leader"));
+        if (ObjectUtils.isEmpty(power)) {
+            //涓嶆槸缁勯暱
+            if (map1.get("look") == 1) {
+                //鏄粍鍛�
+                auxiliaryOutputWorkingHoursDto.setCheck(map1.get("userId"));
+            } else {
+                //绠$悊鍛�(涓嶆坊鍔犻檺鍒舵潯浠舵墍鏈変汉閮藉彲浠ョ湅)
+            }
+        } else {
+            //鏄粍闀�
+            //鏌ヨ缁勯暱涓嬬殑缁勫憳
+            List<User> users = userMapper.selectList(Wrappers.<User>lambdaQuery().like(User::getDepartLimsId, user.getDepartLimsId()));
+            if (ObjectUtils.isNotEmpty(users)) {
+                ids = users.stream().map(User::getId).distinct().collect(Collectors.toList());
+            }
+        }
+        if (ObjectUtils.isNotEmpty(dates)) {
+            String[] split = dates.replaceAll("\\[", "").replaceAll("]", "").replaceAll("\"", "").split(",");
+            if (ObjectUtils.isNotEmpty(week)){
+                String[] weeks = week.replaceAll("\\[", "").replaceAll("]", "").replaceAll("\"", "").split(",");
+                map.put("body", auxiliaryOutputWorkingHoursMapper.selectAuxiliaryOutputWorkingHours(page,
+                        QueryWrappers.queryWrappers(auxiliaryOutputWorkingHoursDto)
+                                .ge("date_time", split[0]).le("date_time", split[1] + " 23:59:59")
+                                .ge("week", weeks[0]).le("week", weeks[1] ), ids));
+            }
+            map.put("body", auxiliaryOutputWorkingHoursMapper.selectAuxiliaryOutputWorkingHours(page, QueryWrappers.queryWrappers(auxiliaryOutputWorkingHoursDto).ge("date_time", split[0]).le("date_time", split[1] + " 23:59:59"), ids));
+        } else {
+            map.put("body", auxiliaryOutputWorkingHoursMapper.selectAuxiliaryOutputWorkingHours(page, QueryWrappers.queryWrappers(auxiliaryOutputWorkingHoursDto), ids));
+        }
         return map;
     }
+
+    //缁熻浜ч噺宸ユ椂姹囨�诲拰杈呭姪宸ユ椂姹囨��
+    @Override
+    public Map<String, Object> collectWorkingHours() {
+        Map<String, Object> map = new HashMap<>();
+        Double sumOutputWorkTime = 0.0;
+        Double sumApprovedWorkingHour = 0.0;
+        List<Integer> ids = null;
+        //鍒ゆ柇鏄粍闀胯繕鏄粍鍛樿繕鏄鐞嗗憳
+        Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectAuxiliaryOutputWorkingHours");
+        User user = userMapper.selectById(map1.get("userId"));
+        Integer roleId = user.getRoleId();
+        Power power = powerMapper.selectOne(Wrappers.<Power>lambdaQuery().eq(Power::getRoleId, roleId).eq(Power::getMenuMethod, "leader"));
+        if (ObjectUtils.isEmpty(power)) {
+            //涓嶆槸缁勯暱
+            if (map1.get("look") == 1) {
+                //鏄粍鍛�
+                ids.add(map1.get("userId"));
+            } else {
+                //绠$悊鍛�(涓嶆坊鍔犻檺鍒舵潯浠舵墍鏈変汉閮藉彲浠ョ湅)
+            }
+        } else {
+            //鏄粍闀�
+            //鏌ヨ缁勯暱涓嬬殑缁勫憳
+            List<User> users = userMapper.selectList(Wrappers.<User>lambdaQuery().like(User::getDepartLimsId, user.getDepartLimsId()));
+            if (ObjectUtils.isNotEmpty(users)) {
+                ids = users.stream().map(User::getId).distinct().collect(Collectors.toList());
+            }
+        }
+        //缁熻褰撳墠鐢ㄦ埛鐨勪骇鍝佸伐鏃�
+        List<AuxiliaryOutputWorkingHours> auxiliaryOutputWorkingHours = auxiliaryOutputWorkingHoursMapper.selectListByIds(ids);
+        if (ObjectUtils.isNotEmpty(auxiliaryOutputWorkingHours)) {
+            Map<String, Double> sumMap = new HashMap<>();
+            for (AuxiliaryOutputWorkingHours auxiliaryOutputWorkingHour : auxiliaryOutputWorkingHours) {
+                if (!sumMap.containsKey(auxiliaryOutputWorkingHour.getManHourGroup())) {
+                    sumMap.put(auxiliaryOutputWorkingHour.getManHourGroup(), auxiliaryOutputWorkingHour.getOutputWorkTime());
+                }
+            }
+            sumOutputWorkTime = sumMap.values().stream().mapToDouble(Double::doubleValue).sum();
+        }
+        map.put("浜ч噺宸ユ椂姹囨��", sumOutputWorkTime);
+        //缁熻褰撳墠鐢ㄦ埛鐨勮緟鍔╁伐鏃�
+        List<AuxiliaryWorkingHoursDay> auxiliaryWorkingHoursDays = auxiliaryWorkingHoursDayMapper.selectListByIds(ids);
+        if (ObjectUtils.isNotEmpty(auxiliaryWorkingHoursDays)) {
+            for (AuxiliaryWorkingHoursDay auxiliaryWorkingHoursDay : auxiliaryWorkingHoursDays) {
+                sumApprovedWorkingHour += auxiliaryWorkingHoursDay.getNonproductiveTime();
+            }
+        }
+        map.put("杈呭姪宸ユ椂姹囨��", sumApprovedWorkingHour);
+        return map;
+    }
+
+    //瀵煎嚭
+    @Override
+    public void exportWorkingHours(HttpServletResponse response) throws IOException {
+        List<AuxiliaryOutputWorkingHoursDto> auxiliaryOutputWorkingHoursDtos = null;
+        List<AuxiliaryWorkingHoursDayDto> auxiliaryWorkingHoursDayDtos = null;
+        List<Integer> ids = null;
+        //鍒ゆ柇鏄粍闀胯繕鏄粍鍛樿繕鏄鐞嗗憳
+        Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectAuxiliaryOutputWorkingHours");
+        User user = userMapper.selectById(map1.get("userId"));
+        Integer roleId = user.getRoleId();
+        Power power = powerMapper.selectOne(Wrappers.<Power>lambdaQuery().eq(Power::getRoleId, roleId).eq(Power::getMenuMethod, "leader"));
+        if (ObjectUtils.isEmpty(power)) {
+            //涓嶆槸缁勯暱
+            if (map1.get("look") == 1) {
+                //鏄粍鍛�
+                ids.add(map1.get("userId"));
+            } else {
+                //绠$悊鍛�(涓嶆坊鍔犻檺鍒舵潯浠舵墍鏈変汉閮藉彲浠ョ湅)
+            }
+        } else {
+            //鏄粍闀�
+            //鏌ヨ缁勯暱涓嬬殑缁勫憳
+            List<User> users = userMapper.selectList(Wrappers.<User>lambdaQuery().like(User::getDepartLimsId, user.getDepartLimsId()));
+            if (ObjectUtils.isNotEmpty(users)) {
+                ids = users.stream().map(User::getId).distinct().collect(Collectors.toList());
+            }
+        }
+
+        //鏌ヨ杈呭姪宸ユ椂
+        auxiliaryWorkingHoursDayDtos = auxiliaryWorkingHoursDayMapper.selectDataByUser(ids);
+//        if (auxiliaryWorkingHoursDayDtos == null || auxiliaryWorkingHoursDayDtos.size() == 0) {
+//            throw new RuntimeException("鏆傛棤鏁版嵁");
+//        }
+        //鏌ヨ缁熻宸ユ椂
+        auxiliaryOutputWorkingHoursDtos = auxiliaryOutputWorkingHoursMapper.selectDataByUser(ids);
+//        if (auxiliaryOutputWorkingHoursDtos == null || auxiliaryOutputWorkingHoursDtos.size() == 0) {
+//            throw new RuntimeException("鏆傛棤鏁版嵁");
+//        }
+        response.setContentType("application/vnd.ms-excel");
+        response.setCharacterEncoding("UTF-8");
+        // 杩欓噷URLEncoder.encode鍙互闃叉涓枃涔辩爜 褰撶劧鍜宔asyexcel娌℃湁鍏崇郴
+        String fileName = URLEncoder.encode("宸ユ椂绠$悊瀵煎嚭", "UTF-8");
+        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
+        try {
+            // 鏂板缓ExcelWriter
+            ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).build();
+
+            WriteSheet mainSheet = EasyExcel.writerSheet(0, "杈呭姪宸ユ椂瀵煎嚭").head(AuxiliaryWorkingHoursDayDto.class).build();
+            excelWriter.write(auxiliaryWorkingHoursDayDtos, mainSheet);
+
+            WriteSheet mainSheet1 = EasyExcel.writerSheet(1, "浜ч噺宸ユ椂瀵煎嚭").head(AuxiliaryOutputWorkingHoursDto.class).build();
+            excelWriter.write(auxiliaryOutputWorkingHoursDtos, mainSheet1);
+            // 鍏抽棴娴�
+            excelWriter.finish();
+        } catch (IOException e) {
+            throw new RuntimeException("瀵煎嚭澶辫触");
+        }
+    }
 }

--
Gitblit v1.9.3