From 1865de1cd0255f7c42a326018a8cc3b5a1ee5253 Mon Sep 17 00:00:00 2001 From: Fixiaobai <fixiaobai@163.com> Date: 星期一, 28 八月 2023 09:03:26 +0800 Subject: [PATCH] 权限 --- user-server/src/main/java/com/yuanchu/limslaboratory/service/impl/UserServiceImpl.java | 67 +++++++++++++++++++++++++-------- 1 files changed, 51 insertions(+), 16 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 794b6ab..9ec7650 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,15 +4,17 @@ 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.MyUtil; import com.yuanchu.limslaboratory.utils.RedisUtil; import com.yuanchu.limslaboratory.pojo.vo.NewPersonnelVo; -import com.yuanchu.limslaboratory.pojo.vo.PagePersonnelVo; import com.yuanchu.limslaboratory.pojo.vo.UpdatePersonnelVo; import org.springframework.stereotype.Service; import org.springframework.util.ObjectUtils; @@ -21,6 +23,8 @@ import java.util.HashMap; import java.util.List; import java.util.Map; +import java.util.Objects; +import java.util.stream.Collectors; /** * <p> @@ -35,6 +39,9 @@ @Resource private UserMapper userMapper; + + @Resource + private RoleManageMapper roleManageMapper; @Override public Map<String, Object> getUserInfo(String token) { @@ -52,7 +59,10 @@ Map<String, Object> data = new HashMap<>(); data.put("name",loginUser.get("name")); data.put("id",loginUser.get("id")); + data.put("roleId",loginUser.get("roleId")); data.put("remind", remind); + RoleAndMenuDto role = getRoleInfo(Long.valueOf(String.valueOf(loginUser.get("roleId")))); + data.put("role",role); return data; } return null; @@ -89,6 +99,11 @@ } @Override + public RoleAndMenuDto getUserRoleInfo(String token) { + return null; + } + + @Override public Map<String, String> selectUserByUserId(int userId) { return userMapper.selectUserByUserId(userId); } @@ -97,7 +112,7 @@ 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); } @@ -114,23 +129,43 @@ @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); + try { + 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 IPage<PagePersonnelVo> getNewPersonnelPage(String name, Page page, Integer status) { - MyUtil.PrintLog(name + status + "============="); + public IPage<Map<String, Object>> getNewPersonnelPage(String name, Page page, Integer status) { return userMapper.getNewPersonnelPage(name, page, status); } + + + public RoleAndMenuDto getRoleInfo(Long roleId) { + 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