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 | 115 +++++++++++++++++++++++++++++++++++++++++++-------------- 1 files changed, 86 insertions(+), 29 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 fc5490c..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 @@ -3,23 +3,28 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; -import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper; 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.vo.NewPersonnelVo; -import com.yuanchu.limslaboratory.vo.PagePersonnelVo; -import com.yuanchu.limslaboratory.vo.UpdatePersonnelVo; +import com.yuanchu.limslaboratory.pojo.vo.NewPersonnelVo; +import com.yuanchu.limslaboratory.pojo.vo.UpdatePersonnelVo; import org.springframework.stereotype.Service; import org.springframework.util.ObjectUtils; import javax.annotation.Resource; import java.util.HashMap; +import java.util.List; import java.util.Map; import java.util.Objects; +import java.util.stream.Collectors; /** * <p> @@ -34,6 +39,9 @@ @Resource private UserMapper userMapper; + + @Resource + private RoleManageMapper roleManageMapper; @Override public Map<String, Object> getUserInfo(String token) { @@ -50,7 +58,11 @@ } 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; @@ -66,6 +78,32 @@ } @Override + public String selectByUserId(Integer userId) { + LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>(); + wrapper.eq(User::getId, userId); + wrapper.select(User::getName); + User user = userMapper.selectOne(wrapper); + return user.getName(); + } + + @Override + public List<Map<String,Object>> selectUser() { + return userMapper.selectUser(); + } + + @Override + public List<Map<String, Object>> getUserNameAndId() { + LambdaQueryWrapper<User> wrapper = new LambdaQueryWrapper<>(); + wrapper.select(User::getId, User::getName); + return userMapper.selectMaps(wrapper); + } + + @Override + public RoleAndMenuDto getUserRoleInfo(String token) { + return null; + } + + @Override public Map<String, String> selectUserByUserId(int userId) { return userMapper.selectUserByUserId(userId); } @@ -74,41 +112,60 @@ 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); + wrapper.select(User::getPassword, User::getRoleId,User::getId, User::getName, User::getAccount, User::getEnterpriseId); return userMapper.selectOne(wrapper); } @Override - public Integer addNewPersonnel(NewPersonnelVo newPersonnelVo) { - User user = new User() - .setName(newPersonnelVo.getName()) - .setAccount(newPersonnelVo.getAccount()) - .setAge(newPersonnelVo.getAge()) - .setPhone(newPersonnelVo.getPhone()) - .setEmail(newPersonnelVo.getEmail()) - .setRoleId(newPersonnelVo.getRole_id()) - .setPassword(newPersonnelVo.getPassword()); - return userMapper.insert(user); + public Integer addNewPersonnel(NewPersonnelVo newPersonnelVo, String enterpriseId) { + try { + User unmarshal = JackSonUtil.unmarshal(JackSonUtil.marshal(newPersonnelVo), User.class); + unmarshal.setEnterpriseId(Integer.valueOf(enterpriseId)); + return userMapper.insert(unmarshal); + } catch (Exception e) { + throw new RuntimeException(e); + } } @Override public Integer updateNewPersonnel(UpdatePersonnelVo updatePersonnelVo) { - User user = new User() - .setName(updatePersonnelVo.getName()) - .setAccount(updatePersonnelVo.getAccount()) - .setAge(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) { - return userMapper.getNewPersonnelPage(name, page); + 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