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