Spring Security 5:为经过OAuth2身份验证的用户提供角色

问题描述

我已经拥有带有Spring Security 5和OAuth2客户端的Spring Boot应用程序,我已经成功配置了使用外部OAuth2提供程序(在我的情况下为GitLab)的身份验证。

现在我在配置授权时遇到问题。我想要一种方法,该方法可以让我编写代码来解析给定用户的角色(通​​过调用数据库或仅检查硬编码的用户名)。

我发现可以通过使用article中描述的PrincipalExtractorAuthoritiesExtractor来实现。但是,这些类在最近的Spring Security中不再存在。 与Spring Security 5兼容的另一种实现方法

解决方法

您要寻找的被称为GrantedAuthoritiesMapper

它已在{spring}官方安全文档中记录here

这是一个代码示例:

@EnableWebSecurity
public class OAuth2LoginSecurityConfig extends WebSecurityConfigurerAdapter {

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http
            .oauth2Login(oauth2 -> oauth2
                .userInfoEndpoint(userInfo -> userInfo
                    .userAuthoritiesMapper(this.userAuthoritiesMapper())
                    ...
                )
            );
    }

    private GrantedAuthoritiesMapper userAuthoritiesMapper() {
        return (authorities) -> {
            Set<GrantedAuthority> mappedAuthorities = new HashSet<>();

            authorities.forEach(authority -> {
                if (OidcUserAuthority.class.isInstance(authority)) {
                    OidcUserAuthority oidcUserAuthority = (OidcUserAuthority)authority;

                    OidcIdToken idToken = oidcUserAuthority.getIdToken();
                    OidcUserInfo userInfo = oidcUserAuthority.getUserInfo();

                    // Map the claims found in idToken and/or userInfo
                    // to one or more GrantedAuthority's and add it to mappedAuthorities

                } else if (OAuth2UserAuthority.class.isInstance(authority)) {
                    OAuth2UserAuthority oauth2UserAuthority = (OAuth2UserAuthority)authority;

                    Map<String,Object> userAttributes = oauth2UserAuthority.getAttributes();

                    // Map the attributes found in userAttributes
                    // to one or more GrantedAuthority's and add it to mappedAuthorities

                }
            });

            return mappedAuthorities;
        };
    }
}

spring安全文档中还有更多示例和解释。