From 40c75db6d335a0335e944e5196a102d8837cad6b Mon Sep 17 00:00:00 2001 From: zss <zss@example.com> Date: 星期一, 15 七月 2024 21:00:28 +0800 Subject: [PATCH] 工时bug修改+光纤接头损耗报告完成 --- performance-server/src/main/java/com/yuanchu/mom/service/impl/AuxiliaryOutputWorkingHoursServiceImpl.java | 190 ++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 187 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..753a514 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,38 @@ 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.ArrayList; import java.util.HashMap; +import java.util.List; import java.util.Map; +import java.util.stream.Collectors; /** * <p> @@ -28,11 +48,175 @@ @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 = new ArrayList<>(); + 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.addAll(users.stream().map(User::getId).distinct().collect(Collectors.toList())); + } + } + if (ids.size()==0){ + ids=null; + } + if (ObjectUtils.isNotEmpty(dates) && ObjectUtils.isNotEmpty(week)){ + String[] split = dates.replaceAll("\\[", "").replaceAll("]", "").replaceAll("\"", "").split(","); + 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)); + }else if(ObjectUtils.isNotEmpty(dates) && ObjectUtils.isEmpty(week)){ + String[] split = dates.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"), ids)); + }else if (ObjectUtils.isEmpty(dates) && ObjectUtils.isNotEmpty(week)){ + String[] weeks = week.replaceAll("\\[", "").replaceAll("]", "").replaceAll("\"", "").split(","); + map.put("body", auxiliaryOutputWorkingHoursMapper.selectAuxiliaryOutputWorkingHours(page, + QueryWrappers.queryWrappers(auxiliaryOutputWorkingHoursDto) + .ge("week", weeks[0]).le("week", weeks[1]), 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 = new ArrayList<>(); + List<AuxiliaryWorkingHoursDayDto> auxiliaryWorkingHoursDayDtos = new ArrayList<>(); + List<Integer> ids = new ArrayList<>(); + //鍒ゆ柇鏄粍闀胯繕鏄粍鍛樿繕鏄鐞嗗憳 + 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); + //鏌ヨ缁熻宸ユ椂 + auxiliaryOutputWorkingHoursDtos = auxiliaryOutputWorkingHoursMapper.selectDataByUser(ids); + 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