操作指南：在 JWT 访问令牌中添加权限作为自定义声明

本指南演示如何将资源所有者权限添加到 JWT 访问令牌。术语“权限”可能代表资源所有者的不同形式，例如角色、权限或组。

为了将资源所有者的权限提供给资源服务器，我们在访问令牌中添加自定义声明。当客户端使用访问令牌访问受保护的资源时，资源服务器将能够获取有关资源所有者访问级别的信息，以及其他潜在用途和好处。

在 JWT 访问令牌中添加自定义声明

您可以使用 OAuth2TokenCustomizer<JWTEncodingContext> @Bean 在访问令牌中添加您自己的自定义声明。请注意，此 @Bean 只能定义一次，因此必须注意确保您正在自定义适当的令牌类型——在本例中为访问令牌。如果您有兴趣自定义 ID 令牌，请参阅 用户资料映射器指南 以获取更多信息。

以下是在访问令牌中添加自定义声明的示例——换句话说，由授权服务器发出的每个访问令牌都将包含自定义声明。

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.security.oauth2.server.authorization.OAuth2TokenType;
import org.springframework.security.oauth2.server.authorization.token.JwtEncodingContext;
import org.springframework.security.oauth2.server.authorization.token.OAuth2TokenCustomizer;

@Configuration
public class CustomClaimsConfiguration {
	@Bean
	public OAuth2TokenCustomizer<JwtEncodingContext> jwtTokenCustomizer() {
		return (context) -> {
			if (OAuth2TokenType.ACCESS_TOKEN.equals(context.getTokenType())) {
				context.getClaims().claims((claims) -> {
					claims.put("claim-1", "value-1");
					claims.put("claim-2", "value-2");
				});
			}
		};
	}
}

在 JWT 访问令牌中添加权限作为自定义声明

要将资源所有者的权限添加到 JWT 访问令牌，我们可以参考上面的自定义声明映射方法，并使用 Principal 的权限填充自定义声明。

我们定义一个具有权限集的示例用户，以供演示目的，并在访问令牌中使用这些权限填充自定义声明。

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.security.core.authority.AuthorityUtils;
import org.springframework.security.core.userdetails.User;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.security.core.userdetails.UserDetailsService;
import org.springframework.security.oauth2.server.authorization.OAuth2TokenType;
import org.springframework.security.oauth2.server.authorization.token.JwtEncodingContext;
import org.springframework.security.oauth2.server.authorization.token.OAuth2TokenCustomizer;
import org.springframework.security.provisioning.InMemoryUserDetailsManager;

import java.util.Collections;
import java.util.Set;
import java.util.stream.Collectors;

@Configuration
public class CustomClaimsWithAuthoritiesConfiguration {
	@Bean
	public UserDetailsService users() {
		UserDetails user = User.withDefaultPasswordEncoder()
				.username("user1") (1)
				.password("password")
				.roles("user", "admin") (2)
				.build();
		return new InMemoryUserDetailsManager(user);
	}

	@Bean
	public OAuth2TokenCustomizer<JwtEncodingContext> jwtTokenCustomizer() { (3)
		return (context) -> {
			if (OAuth2TokenType.ACCESS_TOKEN.equals(context.getTokenType())) { (4)
				context.getClaims().claims((claims) -> { (5)
					Set<String> roles = AuthorityUtils.authorityListToSet(context.getPrincipal().getAuthorities())
							.stream()
							.map(c -> c.replaceFirst("^ROLE_", ""))
							.collect(Collectors.collectingAndThen(Collectors.toSet(), Collections::unmodifiableSet)); (6)
					claims.put("roles", roles); (7)
				});
			}
		};
	}
}
1 使用内存中的 UserDetailsService 定义一个示例用户 user1。
2 为 user1 分配角色。
3 定义一个 OAuth2TokenCustomizer<JwtEncodingContext> @Bean，允许自定义 JWT 声明。
4 检查 JWT 是否为访问令牌。
5 通过 JwtEncodingContext 访问默认声明。
6 从 Principal 对象中提取角色。角色信息存储为以 ROLE_ 为前缀的字符串，因此我们在此处删除前缀。
7 将自定义声明 roles 设置为从上一步收集的角色集。

由于进行了此自定义，有关用户的权限信息将作为自定义声明包含在访问令牌中。