From 124ecca15d3c5d6dcefa9a27aaa3c07c556674e9 Mon Sep 17 00:00:00 2001 From: zss <zss@example.com> Date: 星期二, 30 七月 2024 12:34:42 +0800 Subject: [PATCH] PK8000逻辑处理 --- inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderServiceImpl.java | 158 ++++++++++++++++++++++++++++++++-------------------- 1 files changed, 97 insertions(+), 61 deletions(-) diff --git a/inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderServiceImpl.java b/inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderServiceImpl.java index 576a89a..1b61451 100644 --- a/inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderServiceImpl.java +++ b/inspect-server/src/main/java/com/yuanchu/mom/service/impl/InsOrderServiceImpl.java @@ -2,6 +2,10 @@ import cn.hutool.core.util.ObjectUtil; import cn.hutool.core.util.StrUtil; +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.alibaba.fastjson.JSON; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.core.toolkit.ObjectUtils; @@ -25,6 +29,10 @@ import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; +import javax.servlet.ServletException; +import javax.servlet.http.HttpServletResponse; +import java.io.IOException; +import java.net.URLEncoder; import java.time.LocalDate; import java.time.LocalDateTime; import java.time.format.DateTimeFormatter; @@ -76,10 +84,23 @@ //鑾峰彇妫�楠屼笅鍗曟暟鎹� @Override public Map<String, Object> selectInsOrderParameter(IPage<InsOrder> page, SampleOrderDto sampleOrderDto) { + Map<String, Object> map = new HashMap<>(); + map.put("head", PrintChina.printChina(SampleOrderDto.class)); //鍒ゆ柇鍏ㄩ儴,涓汉,缁勭粐鐨勬潈闄� Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectInsOrderParameter"); - User user = userMapper.selectById(map1.get("userId")); + User user = userMapper.selectById(map1.get("userId"));//褰撳墠鐧诲綍鐨勪汉 Integer roleId = user.getRoleId(); + //鑾峰彇褰撳墠浜烘墍灞炲疄楠屽id + String departLimsId = user.getDepartLimsId(); + String laboratory = null; + if (ObjectUtils.isNotEmpty(departLimsId) && !departLimsId.equals("")) { + String[] split = departLimsId.split(","); + //鏌ヨ瀵瑰簲鏋舵瀯鍚嶇О(閫氫俊瀹為獙瀹�,鐢靛姏瀹為獙瀹�,妫�娴嬪姙) + String departLims = baseMapper.seldepLimsId(Integer.parseInt(split[split.length - 1])); + if (departLims.contains("瀹為獙瀹�")) { + laboratory = departLims; + } + } //鍒ゆ柇鏄惁鏄叏閮ㄦ潈闄� Power power = powerMapper.selectOne(Wrappers.<Power>lambdaQuery().eq(Power::getRoleId, roleId).eq(Power::getMenuMethod, "selectAllInsOrder")); if (ObjectUtils.isEmpty(power)) { @@ -95,15 +116,13 @@ } } } - Map<String, Object> map = new HashMap<>(); - map.put("head", PrintChina.printChina(SampleOrderDto.class)); - map.put("body", insOrderMapper.selectInsOrderPage(page, QueryWrappers.queryWrappers(sampleOrderDto))); + map.put("body", insOrderMapper.selectInsOrderPage(page, QueryWrappers.queryWrappers(sampleOrderDto), laboratory)); return map; } @Override @Transactional(rollbackFor = Exception.class) - public int upInsOrder(Integer orderId, Integer sampleId, String appointed, Integer userId) { + public int upInsOrder(Integer orderId, Integer sampleId, String appointed, Integer userId,String sonLaboratory) { InsOrder insOrder = new InsOrder(); insOrder.setId(orderId); insOrder.setAppointed(LocalDate.parse(appointed)); @@ -128,6 +147,7 @@ insSampleUser.setState(0); insSampleUser.setUserId(userId); insSampleUser.setInsSampleId(orderId); + insSampleUser.setSonLaboratory(sonLaboratory); insSampleUserMapper.insert(insSampleUser); } return 1; @@ -139,7 +159,7 @@ insOrder.setState(0); String code = customMapper.selectById(insOrder.getCompanyId()).getCode2(); //insOrder.setEntrustCode(giveCode.giveCode("JCZX-" + code + "-", "ins_order", "", "yyMMdd")); - String giveCode = this.giveCode.giveCode("JCZX-" + code + "-", "ins_order", "", "yyMMdd"); + String giveCode = this.giveCode.giveCode("JCZX-" + code + "-", "ins_sample", "", "yyMMdd"); insOrderMapper.insert(insOrder); AtomicInteger count = new AtomicInteger(); list.forEach(a -> { @@ -325,6 +345,8 @@ //鍒ゆ柇鐑惊鐜」鐩殑娣诲姞鍜屾俯鍗囪瘯楠岄」鐩殑娣诲姞 else if (product.getInspectionItem().equals("鐑惊鐜�") || product.getInspectionItem().equals("娓╁崌璇曢獙")) { List<InsProduct> insProductess = new ArrayList<>(); + product.setId(null); + product.setInsSampleId(sampleId); insProductess.add(product); List<InsProduct> insProductes = new ArrayList<>(); List<InsProduct> insProducts = new ArrayList<>(); @@ -351,37 +373,29 @@ insProduct.setManHourGroup(product.getManHourGroup()); insProduct.setInspectionItemType("0"); insProduct.setInspectionValueType("1"); + if (product.getInspectionItem().equals("鐑惊鐜�")) { + insProduct.setInspectionItemClass("鐩存祦鐢甸樆");//妫�楠屽瓙瀛愰」--鐩存祦鐢甸樆 + insProduct.setInspectionItemClassEn("DC resistance");//妫�楠屽瓙瀛愰」鑻辨枃--鐩存祦鐢甸樆 + insProduct.setUnit("惟/km");//鍗曚綅 + } insProducts.add(insProduct); insProductes.add(insProduct); insProductess.add(insProduct); } //鐑惊鐜墠鏈夊娆″惊鐜鏁� - if (product.getInspectionItem().equals("鐑惊鐜�")) { - //鐑惊鐜繕瑕佸姞涓婄洿娴佺數闃� - for (InsProduct insProduct : insProducts) { - InsProduct insProduct1 = new InsProduct(); - BeanUtils.copyProperties(insProduct, insProduct1); - insProduct1.setInspectionItemClass("鐩存祦鐢甸樆");//妫�楠屽瓙瀛愰」--鐩存祦鐢甸樆 - insProduct1.setInspectionItemClassEn("DC resistance");//妫�楠屽瓙瀛愰」鑻辨枃--鐩存祦鐢甸樆 - insProduct1.setUnit("惟/km");//鍗曚綅 - insProductes.add(insProduct1); - insProductess.add(insProduct1); - } - } - if (count > 1) { - //寰幆瓒呰繃1娆� - for (int j = 2; j <= count; j++) { - for (InsProduct insProduct : insProductes) { - InsProduct insProduct1 = new InsProduct(); - BeanUtils.copyProperties(insProduct, insProduct1); - insProduct1.setInspectionItem(j + "");//寰幆娆℃暟 - insProductess.add(insProduct1); - } + if (count > 1) { + //寰幆瓒呰繃1娆� + for (int j = 2; j <= count; j++) { + for (InsProduct insProduct : insProductes) { + InsProduct insProduct1 = new InsProduct(); + BeanUtils.copyProperties(insProduct, insProduct1); + insProduct1.setInspectionItem(j + "");//寰幆娆℃暟 + insProductess.add(insProduct1); } } + } insProductService.saveBatch(insProductess); - } - else { + } else { if (!is) { switch (type) { case 1: @@ -462,34 +476,22 @@ map.put("head", PrintChina.printChina(CostStatisticsDto.class)); Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("costStatistics"); if (map1.get("look") == 1) costStatisticsDto.setCreateUser(map1.get("userId")); - IPage<CostStatisticsDto> costStatisticsDtoIPage = insOrderMapper.selectCostStatistics(page, QueryWrappers.queryWrappers(costStatisticsDto).ge("create_time", split[0]).le("create_time", split[1] + " 23:59:59")); - List<CostStatisticsDto> costStatisticsDtoList = costStatisticsDtoIPage.getRecords(); - for (CostStatisticsDto statisticsDto : costStatisticsDtoList) { - statisticsDto.setNum(1); - } - List<CostStatisticsDto> costStatisticsDtoList1 = deduplicateAndSumCounts(costStatisticsDtoList); - costStatisticsDtoIPage.setRecords(costStatisticsDtoList1); - map.put("body", costStatisticsDtoIPage); - return map; - } - private static List<CostStatisticsDto> deduplicateAndSumCounts(List<CostStatisticsDto> list) { - // 浣跨敤涓�涓� Map 鏉ュ瓨鏀惧凡缁忓嚭鐜拌繃鐨� inspectionItem + entrustCode + sample 鐨勭粍鍚� - Map<String, CostStatisticsDto> map = new HashMap<>(); - // 閬嶅巻鍒楄〃杩涜鍘婚噸鍜岃鏁� - for (CostStatisticsDto dto : list) { - String key = dto.getInspectionItem() + dto.getEntrustCode() + dto.getSample(); - if (map.containsKey(key)) { - // 濡傛灉宸茬粡瀛樺湪锛屽垯绱姞鏁伴噺 - CostStatisticsDto existingDto = map.get(key); - existingDto.setNum(existingDto.getNum() + dto.getNum()); - existingDto.setPrice(existingDto.getPrice().add(dto.getPrice())); + IPage<CostStatisticsDto> dtoIPage = insOrderMapper.selectCostStatistics(page, QueryWrappers.queryWrappers(costStatisticsDto).ge("create_time", split[0]).le("create_time", split[1] + " 23:59:59")); + List<CostStatisticsDto> collect = dtoIPage.getRecords().stream().map(dto -> { + Set<String> uniqueTags = new HashSet<>(); + if (dto.getInspectionItem().contains(",")) { + for (String s : dto.getInspectionItem().split(",")) { + uniqueTags.add(s.split("@")[0]); + } } else { - // 濡傛灉涓嶅瓨鍦紝鍒欐斁鍏� Map 涓� - map.put(key, dto); + uniqueTags.add(dto.getInspectionItem().split("@")[0]); } - } - // 灏� Map 涓殑鍊艰浆涓� List 杩斿洖 - return new ArrayList<>(map.values()); + dto.setInspectionItem(uniqueTags.toString()); + return dto; + }).collect(Collectors.toList()); + dtoIPage.setRecords(collect); + map.put("body", dtoIPage); + return map; } @Override @@ -500,16 +502,11 @@ costStatisticsDto.setDates(null); List<CostStatisticsDto> costStatisticsDtos = insOrderMapper.selectCostStatistics2(QueryWrappers.queryWrappers(costStatisticsDto).ge("create_time", split[0]).le("create_time", split[1] + " 23:59:59")); double totalPrice = costStatisticsDtos.stream() - .filter(distinctByKey(dto -> Arrays.asList(dto.getEntrustCode(), dto.getSample(), dto.getCost()))) .filter(dto -> dto.getPrice() != null) // 杩囨护鎺変环鏍间负 null 鐨勫璞� - .mapToDouble(value -> value.getPrice().doubleValue()) + .mapToDouble(value -> value.getPrice().doubleValue() * value.getNum()) .sum(); map.put("total", totalPrice); return map; - } - public static <T> Predicate<T> distinctByKey(Function<? super T, ?> keyExtractor) { - Map<Object, Boolean> seen = new ConcurrentHashMap<>(); - return t -> seen.putIfAbsent(keyExtractor.apply(t), Boolean.TRUE) == null; } @Override @@ -547,6 +544,45 @@ }).collect(Collectors.toList()); return sampleProductDto3s; } + + @Override + public void export(CostStatisticsDto costStatisticsDto,HttpServletResponse response) throws IOException { + //鏌ヨ瀵煎嚭鐨勮垂鐢ㄧ粺璁℃暟鎹� + String dates = costStatisticsDto.getDates(); + String[] split = dates.replaceAll("\\[", "").replaceAll("]", "").replaceAll("\"", "").split(","); + costStatisticsDto.setDates(null); + List<CostStatisticsDto> costStatisticsDtos = insOrderMapper.selectCostStatistics2(QueryWrappers.queryWrappers(costStatisticsDto).ge("create_time", split[0]).le("create_time", split[1] + " 23:59:59")); + costStatisticsDtos = costStatisticsDtos.stream().map(dto -> { + Set<String> uniqueTags = new HashSet<>(); + if (dto.getInspectionItem().contains(",")) { + for (String s : dto.getInspectionItem().split(",")) { + uniqueTags.add(s.split("@")[0]); + } + } else { + uniqueTags.add(dto.getInspectionItem().split("@")[0]); + } + dto.setInspectionItem(uniqueTags.toString()); + return dto; + }).collect(Collectors.toList()); + + 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( "鏍峰搧璐圭敤缁熻瀵煎嚭").head(CostStatisticsDto.class).build(); + excelWriter.write(costStatisticsDtos, mainSheet); + + // 鍏抽棴娴� + excelWriter.finish(); + } catch (IOException e) { + throw new RuntimeException("瀵煎嚭澶辫触"); + } + } } -- Gitblit v1.9.3