From ee2bff3a8eaaafcea4675931db3b8c6ac04c273f Mon Sep 17 00:00:00 2001 From: mitao <2763622819@qq.com> Date: 星期二, 25 二月 2025 15:37:51 +0800 Subject: [PATCH] 修改bug --- springcloud_k8s_panzhihuazhihuishequ/service_sangeshenbian/src/main/java/com/panzhihua/sangeshenbian/api/SystemUserController.java | 25 ++++++++++++++++++------- 1 files changed, 18 insertions(+), 7 deletions(-) diff --git a/springcloud_k8s_panzhihuazhihuishequ/service_sangeshenbian/src/main/java/com/panzhihua/sangeshenbian/api/SystemUserController.java b/springcloud_k8s_panzhihuazhihuishequ/service_sangeshenbian/src/main/java/com/panzhihua/sangeshenbian/api/SystemUserController.java index e02f97b..0500362 100644 --- a/springcloud_k8s_panzhihuazhihuishequ/service_sangeshenbian/src/main/java/com/panzhihua/sangeshenbian/api/SystemUserController.java +++ b/springcloud_k8s_panzhihuazhihuishequ/service_sangeshenbian/src/main/java/com/panzhihua/sangeshenbian/api/SystemUserController.java @@ -2,23 +2,22 @@ import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; -import com.panzhihua.common.constants.SecurityConstants; import com.panzhihua.common.controller.BaseController; import com.panzhihua.common.interfaces.OperLog; -import com.panzhihua.common.model.dtos.PageBaseDTO; import com.panzhihua.common.model.vos.LoginReturnVO; import com.panzhihua.common.model.vos.R; import com.panzhihua.common.model.vos.sangeshenbian.SystemUserVo; import com.panzhihua.common.service.auth.TokenService; -import com.panzhihua.sangeshenbian.model.SystemMenu; -import com.panzhihua.sangeshenbian.model.SystemRoleMenu; -import com.panzhihua.sangeshenbian.model.SystemUser; +import com.panzhihua.sangeshenbian.model.entity.SystemMenu; +import com.panzhihua.sangeshenbian.model.entity.SystemRoleMenu; +import com.panzhihua.sangeshenbian.model.entity.SystemUser; import com.panzhihua.sangeshenbian.service.ISystemMenuService; import com.panzhihua.sangeshenbian.service.ISystemRoleMenuService; import com.panzhihua.sangeshenbian.service.ISystemUserService; import com.panzhihua.sangeshenbian.warpper.LoginVo; import com.panzhihua.sangeshenbian.warpper.SystemUserList; import com.panzhihua.sangeshenbian.warpper.SystemUserListVo; +import com.panzhihua.sangeshenbian.warpper.TokenVo; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; import org.springframework.beans.BeanUtils; @@ -67,7 +66,7 @@ @PostMapping("/login") @ApiOperation(value = "登录", tags = {"三个身边后台-登录"}) @OperLog(operModul = "三个身边后台",operType = 0,businessType = "登录") - public R<LoginReturnVO> login(@RequestBody LoginVo vo){ + public R<TokenVo> login(@RequestBody LoginVo vo){ String key = "login:" + vo.getPhone(); Integer size = (Integer) redisTemplate.opsForValue().get(key); if(null != size && max_err.equals(size)){ @@ -94,7 +93,19 @@ //创建token R<LoginReturnVO> reult = tokenService.loginThreeAround(systemUser.getId()); redisTemplate.delete(key); - return reult; + LoginReturnVO loginReturnVO = reult.getData(); + TokenVo tokenVo = new TokenVo(); + BeanUtils.copyProperties(loginReturnVO, tokenVo); + List<SystemRoleMenu> list = systemRoleMenuService.list(new LambdaQueryWrapper<SystemRoleMenu>().eq(SystemRoleMenu::getSystemRoleId, systemUser.getSystemRoleId())); + List<Integer> collect = list.stream().map(SystemRoleMenu::getSystemMenuId).collect(Collectors.toList()); + Set<String> collect1 = new HashSet<>(); + if(collect.size() > 0){ + List<SystemMenu> systemMenus = systemMenuService.listByIds(collect); + collect1 = systemMenus.stream().map(SystemMenu::getUrl).collect(Collectors.toSet()); + } + tokenVo.setMenu(collect1); + tokenVo.setName(systemUser.getName()); + return R.ok(tokenVo); } -- Gitblit v1.7.1