From ce97969fce786a9a1dc0bc18acad6261bee73a21 Mon Sep 17 00:00:00 2001 From: 李林 <z1292839451@163.com> Date: 星期二, 02 一月 2024 00:39:12 +0800 Subject: [PATCH] 导出功能 --- user-server/src/main/java/com/yuanchu/mom/service/imp/RoleServiceImp.java | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 50 insertions(+), 0 deletions(-) diff --git a/user-server/src/main/java/com/yuanchu/mom/service/imp/RoleServiceImp.java b/user-server/src/main/java/com/yuanchu/mom/service/imp/RoleServiceImp.java index 9edb882..ba28b1b 100644 --- a/user-server/src/main/java/com/yuanchu/mom/service/imp/RoleServiceImp.java +++ b/user-server/src/main/java/com/yuanchu/mom/service/imp/RoleServiceImp.java @@ -2,10 +2,16 @@ import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; +import com.yuanchu.mom.common.GetLook; import com.yuanchu.mom.common.PrintChina; +import com.yuanchu.mom.dto.RolePowerDto; import com.yuanchu.mom.dto.UserPageDto; +import com.yuanchu.mom.mapper.PowerMapper; import com.yuanchu.mom.mapper.RoleMapper; import com.yuanchu.mom.mapper.UserMapper; +import com.yuanchu.mom.pojo.Menu; +import com.yuanchu.mom.pojo.Power; import com.yuanchu.mom.pojo.Role; import com.yuanchu.mom.pojo.User; import com.yuanchu.mom.service.RoleService; @@ -14,6 +20,7 @@ import com.yuanchu.mom.utils.QueryWrappers; import lombok.AllArgsConstructor; import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; import org.springframework.util.DigestUtils; import java.util.HashMap; @@ -26,6 +33,10 @@ private RoleMapper roleMapper; + private PowerMapper powerMapper; + + private GetLook getLook; + @Override public List<Role> selectList() { return roleMapper.selectList(null); @@ -35,6 +46,8 @@ public Map<String, Object> selectUserList(IPage<Role> page, Role role) { Map<String, Object> map = new HashMap<>(); map.put("head", PrintChina.printChina(Role.class)); + Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectRoleLists"); + if(map1.get("look")==1) role.setCreateUser(map1.get("userId")); map.put("body", roleMapper.selectPage(page, QueryWrappers.queryWrappers(role))); return map; } @@ -43,4 +56,41 @@ public int delRole(Integer id) { return roleMapper.deleteById(id); } + + @Override + @Transactional(rollbackFor = Exception.class) + public int addRole(RolePowerDto rolePowerDto) { + Role role = new Role(); + role.setName(rolePowerDto.getRoleName()); + int insert = roleMapper.insert(role); + if (insert == 1){ + rolePowerDto.getPowers().forEach(a->{ + a.setRoleId(role.getId()); + powerMapper.insert(a); + }); + } + return 1; + } + + @Override + public List<Menu> selectMenuList() { + return roleMapper.selectMenuList(); + } + + @Override + @Transactional(rollbackFor = Exception.class) + public int upRole(RolePowerDto rolePowerDto) { + Role role = new Role(); + role.setName(rolePowerDto.getRoleName()); + role.setId(rolePowerDto.getRoleId2()); + int up = roleMapper.updateById(role); + if (up == 1){ + powerMapper.delete(Wrappers.<Power>lambdaUpdate().eq(Power::getRoleId, role.getId())); + rolePowerDto.getPowers().forEach(a->{ + a.setRoleId(role.getId()); + powerMapper.insert(a); + }); + } + return 1; + } } -- Gitblit v1.9.3