diff --git a/src/main/java/com/biutag/supervision/mapper/ConfinementMapper.java b/src/main/java/com/biutag/supervision/mapper/ConfinementMapper.java index 8d28837..ca914a4 100644 --- a/src/main/java/com/biutag/supervision/mapper/ConfinementMapper.java +++ b/src/main/java/com/biutag/supervision/mapper/ConfinementMapper.java @@ -11,6 +11,6 @@ import org.apache.ibatis.annotations.Param; public interface ConfinementMapper extends BaseMapper { - Page queryPage(@Param("page") Page page, @Param(Constants.WRAPPER) QueryWrapper queryWrapper); + Page queryPage(@Param("page") Page page, @Param(Constants.WRAPPER) QueryWrapper queryWrapper, @Param("isNegative") String isNegative); Page queryExcelPage(@Param("page") Page page, @Param(Constants.WRAPPER) QueryWrapper queryWrapper); } diff --git a/src/main/java/com/biutag/supervision/service/ConfinementService.java b/src/main/java/com/biutag/supervision/service/ConfinementService.java index 15df93d..1d00c8b 100644 --- a/src/main/java/com/biutag/supervision/service/ConfinementService.java +++ b/src/main/java/com/biutag/supervision/service/ConfinementService.java @@ -10,21 +10,21 @@ import com.biutag.supervision.constants.AppConstants; import com.biutag.supervision.constants.enums.RoleCodeEnum; import com.biutag.supervision.mapper.ConfinementMapper; import com.biutag.supervision.pojo.entity.Confinement; -import com.biutag.supervision.pojo.entity.NegativeBlame; import com.biutag.supervision.pojo.model.UserAuth; import com.biutag.supervision.pojo.param.ConfinementQueryParam; -import com.biutag.supervision.pojo.param.negativeBlame.NegativeBlameConfinementQueryParam; import com.biutag.supervision.pojo.vo.ConfinementExcelVo; import com.biutag.supervision.pojo.vo.ConfinementVo; import com.biutag.supervision.repository.negativeBlame.NegativeBlameResourceService; import lombok.RequiredArgsConstructor; +import lombok.extern.slf4j.Slf4j; import org.springframework.stereotype.Service; -import java.util.*; -import java.util.stream.Collectors; +import java.util.ArrayList; +import java.util.List; @RequiredArgsConstructor @Service +@Slf4j public class ConfinementService extends ServiceImpl { private final SupDepartService departService; @@ -53,53 +53,55 @@ public class ConfinementService extends ServiceImpl= '2025-04-30T00:00') // if (StrUtil.isNotEmpty(param.getOrgId())) { - if (param.getDepartBranch()) { +// if (param.getDepartBranch()) { List orgIds = departService.getAllNodeIds(param.getOrgId()); queryWrapper.in("c1.org_id", orgIds); - } else { - queryWrapper.in("c1.org_id", param.getOrgId()); - } +// } else { +// queryWrapper.in("c1.org_id", param.getOrgId()); +// } } //todo 排序 - Page confinementVoPage = baseMapper.queryPage(Page.of(param.getCurrent(), param.getSize()), queryWrapper); - List records = confinementVoPage.getRecords(); - if (CollectionUtils.isNotEmpty(records)){ - // 所有的禁闭ID - Set confinementIdList = records.stream().map(ConfinementVo::getId).collect(Collectors.toSet()); - // 去negativeBlame中找negativeId - NegativeBlameConfinementQueryParam negativeBlameQueryParam = new NegativeBlameConfinementQueryParam(); - negativeBlameQueryParam.setConfinementIds(confinementIdList); - negativeBlameQueryParam.setLeadConfinementIds(confinementIdList); - List negativeBlames = negativeBlameResourceService.queryConfinement(negativeBlameQueryParam); - // confinementId --- negativeId - Map negativeIdMap = new HashMap<>(); - for (NegativeBlame nb : negativeBlames) { - if (StrUtil.isNotBlank(nb.getNegativeId())) { - if (StrUtil.isNotBlank(nb.getConfinementId())) { - negativeIdMap.putIfAbsent(nb.getConfinementId(), nb.getNegativeId()); - } - if (StrUtil.isNotBlank(nb.getLeadConfinementId())) { - negativeIdMap.putIfAbsent(nb.getLeadConfinementId(), nb.getNegativeId()); - } - } - } - records.forEach(one -> one.setNegativeId(negativeIdMap.get(one.getId()))); - if (StrUtil.isNotBlank(param.getIsNegative())) { - List filtered = records; - if ("1".equals(param.getIsNegative())) { - filtered = records.stream() - .filter(one -> StrUtil.isNotBlank(one.getNegativeId())) - .collect(Collectors.toList()); - } else if ("0".equals(param.getIsNegative())) { - filtered = records.stream() - .filter(one -> StrUtil.isBlank(one.getNegativeId())) - .collect(Collectors.toList()); - } - confinementVoPage.setRecords(filtered); - confinementVoPage.setTotal(filtered.size()); - } - } - return confinementVoPage; +// Page confinementVoPage = baseMapper.queryPage(Page.of(param.getCurrent(), param.getSize()), queryWrapper, param.getIsNegative()); +// List records = confinementVoPage.getRecords(); +// if (CollectionUtils.isNotEmpty(records)){ +// // 所有的禁闭ID +// Set confinementIdList = records.stream().map(ConfinementVo::getId).collect(Collectors.toSet()); +// // 去negativeBlame中找negativeId +// NegativeBlameConfinementQueryParam negativeBlameQueryParam = new NegativeBlameConfinementQueryParam(); +// negativeBlameQueryParam.setConfinementIds(confinementIdList); +// negativeBlameQueryParam.setLeadConfinementIds(confinementIdList); +// List negativeBlames = negativeBlameResourceService.queryConfinement(negativeBlameQueryParam); +// // confinementId --- negativeId +// Map negativeIdMap = new HashMap<>(); +// for (NegativeBlame nb : negativeBlames) { +// if (StrUtil.isNotBlank(nb.getNegativeId())) { +// if (StrUtil.isNotBlank(nb.getConfinementId())) { +// negativeIdMap.putIfAbsent(nb.getConfinementId(), nb.getNegativeId()); +// } +// if (StrUtil.isNotBlank(nb.getLeadConfinementId())) { +// negativeIdMap.putIfAbsent(nb.getLeadConfinementId(), nb.getNegativeId()); +// } +// } +// } +// records.forEach(one -> one.setNegativeId(negativeIdMap.get(one.getId()))); +// if (StrUtil.isNotBlank(param.getIsNegative())) { +// List filtered = records; +// if ("1".equals(param.getIsNegative())) { +// filtered = records.stream() +// .filter(one -> StrUtil.isNotBlank(one.getNegativeId())) +// .collect(Collectors.toList()); +// log.info("过滤后保留有negativeId的记录: {}", filtered.size()); +// } else if ("0".equals(param.getIsNegative())) { +// filtered = records.stream() +// .filter(one -> StrUtil.isBlank(one.getNegativeId())) +// .collect(Collectors.toList()); +// log.info("过滤后保留有negativeId的记录: {}", filtered.size()); +// } +// confinementVoPage.setRecords(filtered); +// confinementVoPage.setTotal(filtered.size()); +// } +// } + return baseMapper.queryPage(Page.of(param.getCurrent(), param.getSize()), queryWrapper, param.getIsNegative()); } diff --git a/src/main/resources/mapper/ConfinementMapper.xml b/src/main/resources/mapper/ConfinementMapper.xml index 26be6f6..463777e 100644 --- a/src/main/resources/mapper/ConfinementMapper.xml +++ b/src/main/resources/mapper/ConfinementMapper.xml @@ -5,22 +5,73 @@