From 437f8e2b89a18363a1073fdbb3ab99bcd840a757 Mon Sep 17 00:00:00 2001 From: “djh” <“3298565835@qq.com”> Date: 星期一, 21 四月 2025 10:51:36 +0800 Subject: [PATCH] 危化品新版修改 --- hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzStatisticServiceImpl.java | 57 ++++++++++++++++++++++++++++++++++++++++----------------- 1 files changed, 40 insertions(+), 17 deletions(-) diff --git a/hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzStatisticServiceImpl.java b/hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzStatisticServiceImpl.java index c23f010..ebdd109 100644 --- a/hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzStatisticServiceImpl.java +++ b/hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzStatisticServiceImpl.java @@ -51,7 +51,7 @@ private HzWarningMapper warningMapper; @Override - public List<HzEntryRecordVO> entryStatic() { + public List<HzEntryRecordVO> entryStatic(Long companyId) { SysUser currentUser= SecurityUtils.getLoginUser().getUser(); Date currentDate=new Date(); DateTime endTime = DateUtil.endOfDay(currentDate); @@ -68,7 +68,8 @@ dayEntryRecordList.add(entryRecordVO); startTime=DateUtil.offsetDay(startTime,1); } - List<HzEntryRecordVO> entryRecordVOList= entryRecordMapper.entryCountStatic(startDate, endDate,currentUser.getCompanyId()); + //todo + List<HzEntryRecordVO> entryRecordVOList= entryRecordMapper.entryCountStatic(startDate, endDate,currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId()); if(!entryRecordVOList.isEmpty()) { Map<String, HzEntryRecordVO> resMap = entryRecordVOList.stream().collect(Collectors.toMap(item -> item.getMonth()+"_"+item.getDay(), item -> item)); for (HzEntryRecordVO er : dayEntryRecordList) { @@ -83,7 +84,7 @@ } @Override - public List<HzEntryRecordVO> useEverydayStatic() { + public List<HzEntryRecordVO> useEverydayStatic(Long companyId) { SysUser currentUser= SecurityUtils.getLoginUser().getUser(); checkUserAllowed(currentUser); Date currentDate=new Date(); @@ -100,9 +101,14 @@ dayEntryRecordList.add(entryRecordVO); startTime=DateUtil.offsetDay(startTime,1); } - //设置分表id - IdTableNameHandler.setCurrentId(currentUser.getCompanyId()); - List<HzEntryRecordVO> entryRecordVOList= hazmatMapper.useCountEverydayStatic(startDate, endDate,currentUser.getCompanyId()); + if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){ + IdTableNameHandler.setCurrentId(companyId); + }else { + //设置分表id + //todo + IdTableNameHandler.setCurrentId(currentUser.getCompanyId()); + } + List<HzEntryRecordVO> entryRecordVOList= hazmatMapper.useCountEverydayStatic(startDate, endDate,currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId()); IdTableNameHandler.removeCurrentId(); if(!entryRecordVOList.isEmpty()) { Map<String, HzEntryRecordVO> resMap = entryRecordVOList.stream().collect(Collectors.toMap(item -> item.getDay(), item -> item)); @@ -117,7 +123,7 @@ } @Override - public List<HzHazmatUseVO> maxUseStatic() { + public List<HzHazmatUseVO> maxUseStatic(Long companyId) { SysUser currentUser= SecurityUtils.getLoginUser().getUser(); checkUserAllowed(currentUser); @@ -126,9 +132,14 @@ DateTime startTime = DateUtil.beginOfDay(DateUtil.offsetDay(currentDate, -6)); String startDate= DateUtil.format(startTime, DatePattern.NORM_DATETIME_FORMAT); String endDate=DateUtil.format(endTime,DatePattern.NORM_DATETIME_FORMAT); - //设置分表id - IdTableNameHandler.setCurrentId(currentUser.getCompanyId()); - List<HzHazmatUseVO> hazmatUseVOList= hazmatMapper.maxUseCountStatic(startDate, endDate,currentUser.getCompanyId()); + if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){ + IdTableNameHandler.setCurrentId(companyId); + }else { + //设置分表id + //todo + IdTableNameHandler.setCurrentId(currentUser.getCompanyId()); + } + List<HzHazmatUseVO> hazmatUseVOList= hazmatMapper.maxUseCountStatic(startDate, endDate,currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId()); IdTableNameHandler.removeCurrentId(); if(!hazmatUseVOList.isEmpty()) { List<Long> basicIds = hazmatUseVOList.stream().map(HzHazmatUseVO::getBasicId).collect(Collectors.toList()); @@ -145,10 +156,15 @@ } @Override - public HzHomeDataVO homeDataStatistic() { + public HzHomeDataVO homeDataStatistic(Long companyid) { SysUser currentUser=SecurityUtils.getLoginUser().getUser(); checkUserAllowed(currentUser); - Long companyId=currentUser.getCompanyId(); + Long companyId= null; + if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){ + companyId = companyid; + }else { + companyId = currentUser.getCompanyId(); + } Integer warehouseCount= Math.toIntExact(warehouseMapper.selectCount(Wrappers.<HzWarehouse>lambdaQuery() .eq(true, HzWarehouse::getDelFlag, 0) .eq(companyId!=null,HzWarehouse::getCompanyId,companyId))); @@ -156,7 +172,8 @@ stateList.add(HazmatStatusEnum.USED.getCode()); stateList.add(HazmatStatusEnum.DISCARD.getCode()); //设置分表id - IdTableNameHandler.setCurrentId(currentUser.getCompanyId()); + //todo + IdTableNameHandler.setCurrentId(companyId); Integer hazmatCount= Math.toIntExact(hazmatMapper.selectCount(Wrappers.<HzHazmat>lambdaQuery() .notIn(true, HzHazmat::getState, stateList) .eq(companyId!=null,HzHazmat::getCompanyId,companyId))); @@ -178,7 +195,7 @@ @Override - public CommonPage useStatic(String startTime, String endTime) { + public CommonPage useStatic(String startTime, String endTime,Long companyId) { SysUser currentUser= SecurityUtils.getLoginUser().getUser(); checkUserAllowed(currentUser); @@ -201,10 +218,16 @@ throw new ApiException("只能查询90天以内数据"); } } - //设置分表id - IdTableNameHandler.setCurrentId(currentUser.getCompanyId()); + if (currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode())){ + IdTableNameHandler.setCurrentId(companyId); + + }else { + //设置分表id + //todo + IdTableNameHandler.setCurrentId(currentUser.getCompanyId()); + } PageUtils.startPage(); - List<HzHazmatUseVO> hazmatUseVOList= hazmatMapper.useCountStatic(startDate, endDate,currentUser.getCompanyId()); + List<HzHazmatUseVO> hazmatUseVOList= hazmatMapper.useCountStatic(startDate, endDate,currentUser.getUserType().equals(UserTypeEnum.CHECK_USER.getCode()) ? companyId:currentUser.getCompanyId()); IdTableNameHandler.removeCurrentId(); if(!hazmatUseVOList.isEmpty()) { List<Long> basicIds = hazmatUseVOList.stream().map(HzHazmatUseVO::getBasicId).collect(Collectors.toList()); -- Gitblit v1.9.2