shiro与spring security用自定义异常处理401错误

使用shiro与spring security时，我们在认证或授权失败时可能会返回401错误，为了优化用户体验，可以进行自定义异常处理。

下面是使用自定义异常处理401错误的完整攻略。

1. 自定义401异常

我们可以定义一个Custom401Exception类，继承AuthenticationException，并重写构造方法。

public class Custom401Exception extends AuthenticationException {
    public Custom401Exception(String msg, Throwable t) {
        super(msg, t);
    }

    public Custom401Exception(String msg) {
        super(msg);
    }
}

2. 定义异常处理器

定义一个实现了HandlerExceptionResolver接口的异常处理器CustomExceptionHandler，并重写resolveException方法。在此方法中，我们可以判断不同类型的异常并进行处理，例如将Custom401Exception转化为HttpStatus.UNAUTHORIZED响应码。

public class CustomExceptionHandler implements HandlerExceptionResolver {
    @Override
    public ModelAndView resolveException(HttpServletRequest httpServletRequest, HttpServletResponse response, Object o, Exception e) {
        if (e instanceof Custom401Exception) {
            response.setStatus(HttpStatus.UNAUTHORIZED.value());
            return new ModelAndView();
        }
        return null;
    }
}

3. 配置异常处理器

在spring配置文件中，配置自定义异常处理器CustomExceptionHandler。

<bean id="customExceptionHandler" class="com.example.CustomExceptionHandler"/>
<bean class="org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerExceptionResolver">
    <property name="order" value="0"/>
    <property name="defaultErrorView" value="error"/>
    <property name="exceptionResolvers">
        <list>
            <ref bean="customExceptionHandler"/>
        </list>
    </property>
</bean>

4. 抛出自定义异常

在认证或授权失败时，我们可以通过抛出自定义异常Custom401Exception来触发自定义异常处理器。

在shiro中，我们可以通过调用throw new Custom401Exception("Unauthorized");来抛出自定义异常。

在spring security中，我们可以在AuthenticationEntryPoint接口的实现类中，调用response.sendError(HttpServletResponse.SC_UNAUTHORIZED, "Unauthorized");来触发自定义异常。

示例1：shiro中自定义异常处理401错误

public class CustomFilter extends AuthenticatingFilter {

    @Override
    protected AuthenticationToken createToken(ServletRequest servletRequest, ServletResponse servletResponse) throws Exception {
        // 实现创建token的逻辑
    }

    @Override
    protected boolean onAccessDenied(ServletRequest request, ServletResponse response) throws Exception {
        if (isLoginRequest(request, response)) {
            return super.onAccessDenied(request, response);
        } else {
            throw new Custom401Exception("Unauthorized");
        }
    }
}

示例2：spring security中自定义异常处理401错误

public class CustomAuthenticationEntryPoint implements AuthenticationEntryPoint {

    @Override
    public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException authException) throws IOException {
        response.sendError(HttpServletResponse.SC_UNAUTHORIZED, "Unauthorized");
    }
}

以上就是使用shiro与spring security自定义异常处理401错误的攻略，通过自定义异常和异常处理器的组合，我们可以优化用户体验。