sso

src/main/java/org/energy/core/secure/AuthInfo.java

@@ -20,6 +20,9 @@ import com.fasterxml.jackson.databind.ser.std.ToStringSerializer;
 import io.swagger.annotations.ApiModel;
 import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
+import org.energy.modules.system.vo.MenuVO;
+
+import java.util.List;
 
 /**
  * AuthInfo
@@ -54,4 +57,5 @@ public class AuthInfo {
 	private long expiresIn;
 	@ApiModelProperty(value = "许可证")
 	private String license = "powered by daf";
+	private List<MenuVO> roleMenus;
 }

src/main/java/org/energy/modules/auth/controller/AuthController.java

@@ -34,6 +34,7 @@ import org.energy.modules.auth.granter.TokenParameter;
 import org.energy.modules.auth.utils.TokenUtil;
 import org.springframework.web.bind.annotation.*;
 
+import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.TimeUnit;
 
@@ -50,6 +51,7 @@ public class AuthController {
 
 	private RedisUtil redisUtil;
 
+
 	@PostMapping("token")
 	@ApiOperation(value = "获取认证token", notes = "传入租户ID:tenantId,账号:account,密码:password")
 	public R<AuthInfo> token(@ApiParam(value = "授权类型", required = true) @RequestParam(defaultValue = "password", required = false) String grantType,

src/main/java/org/energy/modules/auth/utils/TokenUtil.java

@@ -79,6 +79,7 @@ public class TokenUtil {
 		authInfo.setRefreshToken(createRefreshToken(userInfo).getToken());
 		authInfo.setTokenType(TokenConstant.BEARER);
 		authInfo.setLicense(TokenConstant.LICENSE_NAME);
+		authInfo.setRoleMenus(userInfo.getRoleMenus());
 
 		return authInfo;
 	}

src/main/java/org/energy/modules/system/entity/UserInfo.java

@@ -18,6 +18,7 @@ package org.energy.modules.system.entity;
 import io.swagger.annotations.ApiModel;
 import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
+import org.energy.modules.system.vo.MenuVO;
 
 import java.io.Serializable;
 import java.util.List;
@@ -57,4 +58,10 @@ public class UserInfo implements Serializable {
 	@ApiModelProperty(value = "第三方授权id")
 	private String oauthId;
 
+	/**
+	 * 角色菜单
+	 */
+	@ApiModelProperty(value = "角色菜单")
+	private List<MenuVO> roleMenus;
+
 }

src/main/java/org/energy/modules/system/service/impl/UserServiceImpl.java

@@ -35,6 +35,9 @@ import org.energy.modules.system.mapper.UserMapper;
 import org.energy.modules.system.service.*;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import org.energy.modules.system.service.IMenuService;
+import org.energy.modules.system.vo.MenuVO;
+import com.dayu.daf.core.secure.DafUser;
 
 import java.util.Collections;
 import java.util.List;
@@ -56,6 +59,7 @@ public class UserServiceImpl extends BaseServiceImpl<UserMapper, User> implement
 	private IRoleService roleService;
 	private IUserOauthService userOauthService;
 	private ITenantService tenantService;
+	private IMenuService menuService;
 
 	@Override
 	public boolean submit(User user) {
@@ -95,6 +99,8 @@ public class UserServiceImpl extends BaseServiceImpl<UserMapper, User> implement
 			List<String> roleAlias = baseMapper.getRoleAlias(Func.toLongList(user.getRoleId()));
 			userInfo.setRoles(roleAlias);
 		}
+		List<MenuVO> list = menuService.routes(userInfo.getUser().getRoleId());
+		userInfo.setRoleMenus(list);
 		return userInfo;
 	}
 
@@ -232,6 +238,8 @@ public class UserServiceImpl extends BaseServiceImpl<UserMapper, User> implement
 			List<String> roleAlias = baseMapper.getRoleAlias(Func.toLongList(user.getRoleId()));
 			userInfo.setRoles(roleAlias);
 		}
+		List<MenuVO> list = menuService.routes(userInfo.getUser().getRoleId());
+		userInfo.setRoleMenus(list);
 		return userInfo;
 	}
 }