Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MeterSphere
metersphere
提交
182d5b24
M
metersphere
项目概览
MeterSphere
/
metersphere
上一次同步 3 年多
通知
25
Star
1
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
metersphere
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
182d5b24
编写于
2月 19, 2020
作者:
C
Captain.B
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
set roles
上级
14a6b787
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
15 addition
and
3 deletion
+15
-3
backend/src/main/java/io/metersphere/security/ShiroDBRealm.java
...d/src/main/java/io/metersphere/security/ShiroDBRealm.java
+15
-3
未找到文件。
backend/src/main/java/io/metersphere/security/ShiroDBRealm.java
浏览文件 @
182d5b24
package
io.metersphere.security
;
import
io.metersphere.base.domain.Role
;
import
io.metersphere.dto.UserDTO
;
import
io.metersphere.service.UserService
;
import
io.metersphere.user.SessionUser
;
import
org.apache.shiro.SecurityUtils
;
import
org.apache.shiro.authc.*
;
import
org.apache.shiro.authz.AuthorizationInfo
;
import
org.apache.shiro.authz.SimpleAuthorizationInfo
;
import
org.apache.shiro.realm.AuthorizingRealm
;
import
org.apache.shiro.subject.PrincipalCollection
;
import
org.slf4j.Logger
;
import
org.slf4j.LoggerFactory
;
import
javax.annotation.Resource
;
import
java.util.Set
;
import
java.util.stream.Collectors
;
/**
...
...
@@ -34,8 +38,16 @@ public class ShiroDBRealm extends AuthorizingRealm {
* 权限认证
*/
@Override
protected
AuthorizationInfo
doGetAuthorizationInfo
(
PrincipalCollection
principalCollection
)
{
return
null
;
protected
AuthorizationInfo
doGetAuthorizationInfo
(
PrincipalCollection
principals
)
{
String
userName
=
(
String
)
principals
.
getPrimaryPrincipal
();
SimpleAuthorizationInfo
authorizationInfo
=
new
SimpleAuthorizationInfo
();
// roles 内容填充
UserDTO
userDTO
=
userService
.
getUserDTO
(
userName
);
Set
<
String
>
roles
=
userDTO
.
getRoles
().
stream
().
map
(
Role:
:
getId
).
collect
(
Collectors
.
toSet
());
authorizationInfo
.
setRoles
(
roles
);
return
authorizationInfo
;
}
/**
...
...
@@ -53,7 +65,7 @@ public class ShiroDBRealm extends AuthorizingRealm {
logger
.
warn
(
msg
);
throw
new
UnknownAccountException
(
msg
);
}
// TODO 密码验证
,roles 等内容填充
// TODO 密码验证
SessionUser
sessionUser
=
SessionUser
.
fromUser
(
user
);
SecurityUtils
.
getSubject
().
getSession
().
setAttribute
(
"user"
,
sessionUser
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录