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