From 7ef4892f9f24f941aca37e6b3991b808a0aca619 Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期五, 08 九月 2023 11:16:35 +0800
Subject: [PATCH] 优化

---
 ycl-common/src/main/java/com/ycl/service/user/impl/UmsRoleServiceImpl.java |   12 +++++++++---
 1 files changed, 9 insertions(+), 3 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 fef2b37..9f060c4 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
@@ -8,7 +8,7 @@
 import com.ycl.dto.UmsMenuNode;
 import com.ycl.entity.dict.DataDictionary;
 import com.ycl.entity.user.*;
-import com.ycl.mapper.dict.DataDictionaryMapper;
+import com.ycl.mapper.dict.DataDictionary2Mapper;
 import com.ycl.mapper.user.UmsMenuMapper;
 import com.ycl.mapper.user.UmsResourceMapper;
 import com.ycl.mapper.user.UmsRoleMapper;
@@ -47,7 +47,7 @@
     @Resource
     private UmsResourceMapper umsResourceMapper;
     @Resource
-    private DataDictionaryMapper dataDictionaryMapper;
+    private DataDictionary2Mapper dataDictionary2Mapper;
 
     @Override
     public boolean create(UmsRole role) {
@@ -78,7 +78,7 @@
         if (types.size() > 0) {
             LambdaQueryWrapper<DataDictionary> dicQuery = new LambdaQueryWrapper<>();
             dicQuery.in(DataDictionary::getId, types);
-            List<DataDictionary> dics = dataDictionaryMapper.selectList(dicQuery);
+            List<DataDictionary> dics = dataDictionary2Mapper.selectList(dicQuery);
             for (UmsRole role : roles) {
                 for (DataDictionary dic : dics) {
                     if (dic.getId().equals(role.getType())) {
@@ -101,6 +101,12 @@
 
 
     @Override
+    public List<UmsRole> getRoleListByCodes(List<String> codes) {
+        LambdaQueryWrapper<UmsRole> wrapper = new LambdaQueryWrapper<UmsRole>().in(UmsRole::getCode, codes);
+        return baseMapper.selectList(wrapper);
+    }
+
+    @Override
     public List<UmsMenu> listMenu(Long roleId) {
         return umsMenuMapper.getMenuListByRoleId(roleId);
     }

--
Gitblit v1.8.0