From b6373daa40c3dc17ca107f3666fd252f3b6b0778 Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期五, 03 一月 2025 18:34:52 +0800 Subject: [PATCH] 代码 --- ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TIndexMenuServiceImpl.java | 19 +++++++++++++------ 1 files changed, 13 insertions(+), 6 deletions(-) diff --git a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TIndexMenuServiceImpl.java b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TIndexMenuServiceImpl.java index 335afec..8da223a 100644 --- a/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TIndexMenuServiceImpl.java +++ b/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TIndexMenuServiceImpl.java @@ -1,5 +1,6 @@ package com.ruoyi.system.service.impl; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; @@ -14,9 +15,8 @@ import org.springframework.stereotype.Service; import java.time.LocalDate; -import java.util.Collections; -import java.util.List; -import java.util.Map; +import java.util.*; +import java.util.stream.Collectors; /** * <p> @@ -34,10 +34,17 @@ @Override public List<AllertTitleDto> allert(List<String> cityCode) { - List<AllertTitleDto> allert = this.baseMapper.allert(cityCode); + List<AllertTitleDto> allert = new ArrayList<>(); + List<String> strings = new ArrayList<>(); + for (String s : cityCode) { + strings.add(s); + } + List<AllertTitleDto> allert1 = this.baseMapper.allert(strings); + allert.addAll(allert1); + allert = allert.stream().distinct().collect(Collectors.toList()); + List<TTitleMajor> tTitleMajorList = titleMajorMapper.selectList(new LambdaQueryWrapper<TTitleMajor>()); for (AllertTitleDto allertTitleDto : allert) { - List<TTitleMajor> tTitleMajors = titleMajorMapper.selectList(Wrappers.lambdaQuery(TTitleMajor.class) - .eq(TTitleMajor::getTechnicalId, allertTitleDto.getId())); + List<TTitleMajor> tTitleMajors = tTitleMajorList.stream().filter(e -> e.getTechnicalId().equals(allertTitleDto.getId())).collect(Collectors.toList()); allertTitleDto.setChildren(tTitleMajors); } return allert; -- Gitblit v1.7.1