我正在编写Spring Boot REST API,并且正在使用JWT令牌。现在,我正在尝试创建基于角色的授权。
这是我正在使用的教程/实现。
https://auth0.com/blog/implementing-jwt-authentication-on-spring-boot/
我通过附加的Role实体扩展了此实现,并将@ManyToMany映射添加到ApplicationUser Entity。
现在,据我所知,应该在令牌创建过程中将用户角色添加到令牌中。 因此,这是一个现有代码:
@Override
protected void successfulAuthentication(HttpServletRequest req, HttpServletResponse res, FilterChain chain, Authentication auth) throws IOException, ServletException {
String token = JWT.create()
.withSubject(((User) auth.getPrincipal()).getUsername())
.withExpiresAt(new Date(System.currentTimeMillis() + EXPIRATION_TIME))
.sign(HMAC512(SECRET.getBytes()));
res.addHeader(HEADER_STRING, TOKEN_PREFIX + token);
}
我猜应该在此处添加用户角色。有一个功能:
withArrayClaim(String Name, String[] items)
第一个问题是:我不确定如何正确添加它。
然后,这是片段,据我了解,这是验证令牌的地方:
private UsernamePasswordAuthenticationToken getAuthentication(HttpServletRequest request) {
String token = request.getHeader(HEADER_STRING);
if (token != null) {
// parse the token.
String user = JWT.require(Algorithm.HMAC512(SECRET.getBytes()))
.build()
.verify(token.replace(TOKEN_PREFIX, ""))
.getSubject();
if (user != null) {
return new UsernamePasswordAuthenticationToken(user, null, new ArrayList<>());
}
return null;
}
return null;
}
困扰我的是片段:
return new UsernamePasswordAuthenticationToken(user, null, new ArrayList<>());
我不明白为什么会有空值(Inteliij将其突出显示为“凭据”)和这个新的ArrayList。有没有地方我应该从令牌中获取角色并添加它们?
我知道,这是一个范围很广的问题,但是我找不到其他解决方案。 或者也许有一种更简单的方法来创建简单的JWT令牌认证/授权(基于角色)。
期待您的回答!
编辑:
或者也许还有比其他更简单的解决方案-不将用户角色保留在键内-而是仅将它们添加到null和new ArrayList所在的“第二部分”中?
答案 0 :(得分:1)
只需根据用户角色创建授予的权限,然后对其进行身份验证。然后,经过身份验证的用户主体将包含角色。 简单的例子:
UserEntity userEntity = userRepository.findUserByEmail(user); // this depends of course on your implementation
if (userEntity == null) return null;
List<RoleEntity> roles = userEntity.getRoles();
Collection<GrantedAuthority> authorities = new HashSet<>();
roles.forEach((role) -> {
authorities.add(new SimpleGrantedAuthority(role.getName()));
});
return new UsernamePasswordAuthenticationToken(user, null, authorities);
更好的是,您可以创建一个实现来自Spring安全性的UserDetails的UserPrincipal。
public class UserPrincipal implements UserDetails {
private static final long serialVersionUID = 1L;
private final UserEntity userEntity;
public UserPrincipal(UserEntity userEntity){
this.userEntity = userEntity;
}
@Override
public Collection<? extends GrantedAuthority> getAuthorities() {
Collection<GrantedAuthority> authorities = new HashSet<>();
// Get user Roles
Collection<RoleEntity> roles = userEntity.getRoles();
if(roles == null) return authorities;
roles.forEach((role) -> {
authorities.add(new SimpleGrantedAuthority(role.getName()));
});
return authorities;
}
@Override
public String getPassword() {
return this.userEntity.getEncryptedPassword();
}
@Override
public String getUsername() {
return this.userEntity.getEmail();
}
@Override
public boolean isAccountNonExpired() {
return false;
}
@Override
public boolean isAccountNonLocked() {
return false;
}
@Override
public boolean isCredentialsNonExpired() {
return false;
}
@Override
public boolean isEnabled() {
return false;
}
}
并使用它:
UserEntity userEntity = userRepository.findUserByEmail(user);
if (userEntity == null) return null;
UserPrincipal userPrincipal = new UserPrincipal(userEntity);
return new UsernamePasswordAuthenticationToken(userPrincipal, null, userPrincipal.getAuthorities());