From 5d9f3637b82a99aa9d9d4154a42501599dba4536 Mon Sep 17 00:00:00 2001
From: mg <maokecheng@163.com>
Date: 星期三, 19 十月 2022 14:43:13 +0800
Subject: [PATCH] 修改角色类型bug。

---
 ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java |   21 ++++++++++++++++++++-
 1 files changed, 20 insertions(+), 1 deletions(-)

diff --git a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java
index 2dae3bd..d610315 100644
--- a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java
+++ b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java
@@ -5,7 +5,9 @@
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.ycl.entity.dict.DataDictionary;
 import com.ycl.entity.user.*;
+import com.ycl.mapper.dict.DataDictionaryMapper;
 import com.ycl.mapper.user.UmsMenuMapper;
 import com.ycl.mapper.user.UmsResourceMapper;
 import com.ycl.mapper.user.UmsRoleMapper;
@@ -20,6 +22,7 @@
 import java.util.ArrayList;
 import java.util.Date;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * 鍚庡彴瑙掕壊绠$悊Service瀹炵幇绫�
@@ -37,6 +40,8 @@
     private UmsMenuMapper umsMenuMapper;
     @Resource
     private UmsResourceMapper umsResourceMapper;
+    @Resource
+    private DataDictionaryMapper dataDictionaryMapper;
     @Override
     public boolean create(UmsRole role) {
         role.setCreateTime(new Date());
@@ -60,7 +65,21 @@
         if(StrUtil.isNotEmpty(keyword)){
             lambda.like(UmsRole::getName,keyword);
         }
-        return page(page,wrapper);
+        Page<UmsRole> resultPage = page(page, wrapper);
+        List<UmsRole> roles = resultPage.getRecords();
+        List<Long> types = roles.stream().map(UmsRole::getType).collect(Collectors.toList());
+
+        LambdaQueryWrapper<DataDictionary> dicQuery = new LambdaQueryWrapper<>();
+        dicQuery.in(DataDictionary::getId, types);
+        List<DataDictionary> dics = dataDictionaryMapper.selectList(dicQuery);
+        for (UmsRole role : roles) {
+            for (DataDictionary dic : dics) {
+                if (dic.getId().equals(role.getType())) {
+                    role.setTypeText(dic.getName());
+                }
+            }
+        }
+        return resultPage;
     }
 
     @Override

--
Gitblit v1.8.0