From 0a1533fd30ec1a2f4624ccda4ff11f2535ea8a46 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期三, 11 十二月 2024 18:37:14 +0800
Subject: [PATCH] 修改

---
 ruoyi-system/src/main/java/com/ruoyi/system/service/impl/TIndexMenuServiceImpl.java |   24 ++++++++++++++++++++----
 1 files changed, 20 insertions(+), 4 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 90fea1f..d7eac65 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,17 +1,20 @@
 package com.ruoyi.system.service.impl;
 
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 
 import com.ruoyi.system.domain.TIndexMenu;
+import com.ruoyi.system.domain.TTitleMajor;
 import com.ruoyi.system.dto.AllertTitleDto;
 import com.ruoyi.system.mapper.TIndexMenuMapper;
+import com.ruoyi.system.mapper.TTitleMajorMapper;
 import com.ruoyi.system.service.TIndexMenuService;
+import com.ruoyi.system.service.TTitleMajorService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.time.LocalDate;
-import java.util.Collections;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * <p>
@@ -24,9 +27,22 @@
 @Service
 public class TIndexMenuServiceImpl extends ServiceImpl<TIndexMenuMapper, TIndexMenu> implements TIndexMenuService {
 
+    @Autowired
+    private TTitleMajorMapper titleMajorMapper;
+
     @Override
     public List<AllertTitleDto> allert(List<String> cityCode) {
-        return this.baseMapper.allert(cityCode);
+        List<AllertTitleDto> allert = new ArrayList<>();
+        for (String s : cityCode) {
+            List<AllertTitleDto> allert1 = this.baseMapper.allert(Collections.singletonList(s));
+            allert.addAll(allert1);
+        }
+        for (AllertTitleDto allertTitleDto : allert) {
+            List<TTitleMajor> tTitleMajors = titleMajorMapper.selectList(Wrappers.lambdaQuery(TTitleMajor.class)
+                    .eq(TTitleMajor::getTechnicalId, allertTitleDto.getId()));
+            allertTitleDto.setChildren(tTitleMajors);
+        }
+        return allert;
     }
 
     @Override

--
Gitblit v1.7.1