基于角色的登录

本文展示了如何根据不同的用户角色，在登录之后来重定向到不同的页面。

在 method-security项目的基础上，我们构建了一个role-base-login项目。

build.gradle

修改 build.gradle 文件，让我们的role-base-login项目成为一个新的项目。

修改内容也比较简单，修改项目名称及版本即可。

jar {
    baseName = 'role-base-login'
    version = '1.0.0'
}

权限处理器

新增com.waylau.spring.boot.security.auth包，用于存放权限处理器相关的类。

新建 AuthenticationSuccessHandler ，继承自 org.springframework.security.web.authentication.SimpleUrlAuthenticationSuccessHandler

@Component
public class AuthenticationSuccessHandler extends SimpleUrlAuthenticationSuccessHandler {
    private RedirectStrategy redirectStrategy = new DefaultRedirectStrategy();

    @Override
    public void handle(HttpServletRequest request, HttpServletResponse response,
            Authentication authentication) throws IOException, ServletException {
        String targetUrl = determineTargetUrl(authentication);

        if (response.isCommitted()) {
            logger.debug("Response has already been committed. Unable to redirect to "
                    + targetUrl);
            return;
        }

        redirectStrategy.sendRedirect(request, response, targetUrl);
    }

    private String determineTargetUrl(Authentication authentication) {
        String targetUrl = null;

        Collection<? extends GrantedAuthority> authorities = authentication.getAuthorities();

        List<String> roles = new ArrayList<String>();

        for (GrantedAuthority a : authorities) {
            roles.add(a.getAuthority());
        }

        // 根据不同的角色，重定向到不同页面
        if (isAdmin(roles)) {
            targetUrl = "/admins";
        } else if (isUser(roles)) {
            targetUrl = "/users";
        } else {
            targetUrl = "/403";
        }

        return targetUrl;
    }

    private boolean isUser(List<String> roles) {
        if (roles.contains("ROLE_USER")) {
            return true;
        }
        return false;
    }

    private boolean isAdmin(List<String> roles) {
        if (roles.contains("ROLE_ADMIN")) {
            return true;
        }
        return false;
    }

}

重写了 handle()方法，这个方法简单地使用 RedirectStrategy 来重定向，由自定义的 determineTargetUrl 方法返回URL。此方法提取当前认证对象用户记录的角色，然后构造基于角色有相应的URL。最后由负责 Spring Security 框架内的所有重定向的 RedirectStrategy ，将请求重定向到指定的URL。

修改配置类

在配置类中，添加成功认证的处理器 AuthenticationSuccessHandler 的实例：

.formLogin()   //基于 Form 表单登录验证
    .loginPage("/login").failureUrl("/login-error") // 自定义登录界面
    .successHandler(authenticationSuccessHandler)

运行

当我们使用“ADMIN”角色的用户登录成功后，会被重定向到/admins页面。而使用“USER”角色的用户登录成功后，会被重定向到/users页面。