From dd2554435b9ced61e2a6a06627145fca3bf2685b Mon Sep 17 00:00:00 2001
From: Fixiaobai <fixiaobai@163.com>
Date: 星期一, 04 九月 2023 17:17:17 +0800
Subject: [PATCH] 修改bug

---
 user-server/src/main/java/com/yuanchu/limslaboratory/service/impl/UserServiceImpl.java |   85 +++++++++++++++++++++++++++++++-----------
 1 files changed, 62 insertions(+), 23 deletions(-)

diff --git a/user-server/src/main/java/com/yuanchu/limslaboratory/service/impl/UserServiceImpl.java b/user-server/src/main/java/com/yuanchu/limslaboratory/service/impl/UserServiceImpl.java
index f18693f..47b0741 100644
--- a/user-server/src/main/java/com/yuanchu/limslaboratory/service/impl/UserServiceImpl.java
+++ b/user-server/src/main/java/com/yuanchu/limslaboratory/service/impl/UserServiceImpl.java
@@ -4,11 +4,15 @@
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.yuanchu.limslaboratory.mapper.RoleManageMapper;
+import com.yuanchu.limslaboratory.pojo.RoleMenu;
 import com.yuanchu.limslaboratory.pojo.User;
 import com.yuanchu.limslaboratory.mapper.UserMapper;
+import com.yuanchu.limslaboratory.pojo.dto.RoleAndMenuDto;
 import com.yuanchu.limslaboratory.service.UserService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yuanchu.limslaboratory.utils.JackSonUtil;
+import com.yuanchu.limslaboratory.utils.JsonUtil;
 import com.yuanchu.limslaboratory.utils.MyUtil;
 import com.yuanchu.limslaboratory.utils.RedisUtil;
 import com.yuanchu.limslaboratory.pojo.vo.NewPersonnelVo;
@@ -20,6 +24,8 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -34,6 +40,9 @@
 
     @Resource
     private UserMapper userMapper;
+
+    @Resource
+    private RoleManageMapper roleManageMapper;
 
     @Override
     public Map<String, Object> getUserInfo(String token) {
@@ -50,8 +59,13 @@
             }
             Map<String, Object> data = new HashMap<>();
             data.put("name",loginUser.get("name"));
+            data.put("account", loginUser.get("account"));
             data.put("id",loginUser.get("id"));
+            data.put("roleId",loginUser.get("roleId"));
             data.put("remind", remind);
+            data.put("sessionLayerId",loginUser.get("sessionLayerId"));
+            RoleAndMenuDto role = getRoleInfo(Long.valueOf(String.valueOf(loginUser.get("roleId"))));
+            data.put("role",role);
             return data;
         }
         return null;
@@ -81,10 +95,16 @@
     }
 
     @Override
-    public List<Map<String, Object>> getUserNameAndId() {
+    public List<Map<String, Object>>
+    getUserNameAndId() {
         LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>();
-        wrapper.select(User::getId, User::getName);
+        wrapper.select(User::getId, User::getName).eq(User::getType,1);
         return userMapper.selectMaps(wrapper);
+    }
+
+    @Override
+    public RoleAndMenuDto getUserRoleInfo(String token) {
+        return null;
     }
 
     @Override
@@ -96,40 +116,59 @@
     public User AccordingUsernameSelectAll(String account) {
         LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>();
         wrapper.eq(User::getAccount, account);
-        wrapper.select(User::getPassword, User::getId, User::getName, User::getAccount, User::getEnterpriseId);
+        wrapper.select(User::getPassword, User::getRoleId,User::getId, User::getName, User::getAccount, User::getEnterpriseId);
         return userMapper.selectOne(wrapper);
     }
 
     @Override
     public Integer addNewPersonnel(NewPersonnelVo newPersonnelVo, String enterpriseId) {
+        User unmarshal = JsonUtil.jsonToPojo(JsonUtil.jsonToString(newPersonnelVo), User.class);
+        unmarshal.setEnterpriseId(Integer.valueOf(enterpriseId));
+        return userMapper.insert(unmarshal);
+    }
+
+    @Override
+    public Integer updateNewPersonnel(UpdatePersonnelVo updatePersonnelVo) {
         try {
-            User unmarshal = JackSonUtil.unmarshal(JackSonUtil.marshal(newPersonnelVo), User.class);
-            unmarshal.setEnterpriseId(Integer.valueOf(enterpriseId));
-            return userMapper.insert(unmarshal);
+            MyUtil.PrintLog(updatePersonnelVo.toString());
+            User unmarshal = JackSonUtil.unmarshal(JackSonUtil.marshal(updatePersonnelVo), User.class);
+
+            LambdaUpdateWrapper<User> updateWrapper = new LambdaUpdateWrapper<>();
+            updateWrapper.eq(User::getId, updatePersonnelVo.getId());
+            return userMapper.update(unmarshal, updateWrapper);
         } catch (Exception e) {
             throw new RuntimeException(e);
         }
     }
 
     @Override
-    public Integer updateNewPersonnel(UpdatePersonnelVo updatePersonnelVo) {
-        User user = new User()
-                .setName(updatePersonnelVo.getName())
-                .setAccount(updatePersonnelVo.getAccount())
-                .setAge(String.valueOf(updatePersonnelVo.getAge()))
-                .setPhone(updatePersonnelVo.getPhone())
-                .setEmail(updatePersonnelVo.getEmail())
-                .setRoleId(updatePersonnelVo.getRole_id())
-                .setPassword(updatePersonnelVo.getPassword())
-                .setJobState(updatePersonnelVo.getJobState());
-        LambdaUpdateWrapper<User> updateWrapper = new LambdaUpdateWrapper<>();
-        updateWrapper.eq(User::getId, updatePersonnelVo.getId());
-        return userMapper.update(user, updateWrapper);
+    public IPage<Map<String, Object>> getNewPersonnelPage(String name, Page page, Integer status) {
+        return userMapper.getNewPersonnelPage(name, page, status);
     }
 
-    @Override
-    public IPage<Map<String, Object>> getNewPersonnelPage(String name, Page page, Integer status) {
-        MyUtil.PrintLog(name + status + "=============");
-        return userMapper.getNewPersonnelPage(name, page, status);
+
+    public RoleAndMenuDto getRoleInfo(Long roleId) {
+        if(roleId==0){
+            return null;
+        }
+        RoleAndMenuDto roleAndMenuDto = roleManageMapper.selectAllRoleAndMenuByRoleId(roleId);
+        List<RoleMenu> collect = roleAndMenuDto.getRoleMenuList()
+                .stream()
+                .filter(t -> t.getParentId() == 0)
+                .peek((menu) -> menu.setChildren(this.getRoleMenuChildren(menu, roleAndMenuDto.getRoleMenuList())))
+                .collect(Collectors.toList());
+        roleAndMenuDto.setRoleMenuList(collect);
+        return roleAndMenuDto;
+    }
+
+
+    private List<RoleMenu> getRoleMenuChildren(RoleMenu root, List<RoleMenu> all) {
+        return all.stream()
+                .filter(t -> Objects.equals(t.getParentId(), root.getMenuId()))
+                .peek(g -> {
+                    //鎵惧瓙鑿滃崟
+                    g.setChildren(getRoleMenuChildren(g, all));
+                })
+                .collect(Collectors.toList());
     }
 }

--
Gitblit v1.9.3