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

我将为您详细讲解 “Shiro与Spring Security用自定义异常处理401错误”的完整攻略。

首先，我们先了解一下什么是401错误。401错误表示未经授权或身份验证失败。在Shiro和Spring Security中，当用户获取未授权的访问时，系统将返回401错误。

接着，我们可以通过自定义异常处理程序来处理401错误。

一、Shiro的自定义异常处理401错误

1.首先在Shiro配置文件中添加异常处理器

<!-- 添加自定义异常处理器 -->
<bean id="exceptionHandler" class="com.example.MyExceptionHandler"></bean>
<property name="exceptionHandler" ref="exceptionHandler"/>

2.编写自定义异常处理器MyExceptionHandler.java

public class MyExceptionHandler extends SimpleMappingExceptionResolver {
    @Override
    protected ModelAndView doResolveException(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {

        if (ex instanceof UnauthorizedException) {
            response.setStatus(401);
            return new ModelAndView("redirect:/401");
        }
        return super.doResolveException(request, response, handler, ex);
    }
}

这里我们重写了SimpleMappingExceptionResolver的doResolveException方法，当异常是UnauthorizedException时，设置响应状态码为401，跳转到指定的401页面。如果是其他异常，则执行默认的处理方式。

3.在web.xml文件中配置ShiroFilter

<!-- 配置ShiroFilter -->
<filter>
    <filter-name>ShiroFilter</filter-name>
    <filter-class>org.springframework.web.filter.DelegatingFilterProxy</filter-class>
</filter>

<filter-mapping>
    <filter-name>ShiroFilter</filter-name>
    <url-pattern>/*</url-pattern>
</filter-mapping>

二、Spring Security的自定义异常处理401错误

1.继承Spring Boot的ErrorController类，并实现处理方法

@RestController
public class MyErrorController implements ErrorController {

    private static final String PATH = "/error";

    @RequestMapping(PATH)
    public ResponseEntity error(HttpServletRequest request) {

        HttpStatus status = getStatus(request);
        if (status == HttpStatus.UNAUTHORIZED) {
            return ResponseEntity.status(HttpStatus.UNAUTHORIZED).body("Unauthorized");
        }
        return ResponseEntity.status(status).body(status.getReasonPhrase());
    }

    private HttpStatus getStatus(HttpServletRequest request) {
        Integer statusCode = (Integer) request.getAttribute(RequestDispatcher.ERROR_STATUS_CODE);
        if (statusCode != null) {
            try {
                return HttpStatus.valueOf(statusCode);
            }
            catch (Exception ex) {
                // ignore
            }
        }
        return HttpStatus.INTERNAL_SERVER_ERROR;
    }

    @Override
    public String getErrorPath() {
        return PATH;
    }
}

2.在Spring Boot应用程序主类中添加错误处理器

@SpringBootApplication
public class Application {
    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }

    @Bean
    public WebMvcConfigurer webMvcConfigurer() {
        return new WebMvcConfigurerAdapter() {
            @Override
            public void addViewControllers(ViewControllerRegistry registry) {
                registry.addViewController("/401").setViewName("401");
            }

            @Override
            public void configureErrorControllers(ErrorPageRegistry registry) {
                registry.addErrorController(new MyErrorController());
            }
        };
    }
}

这里我们添加了一个自定义错误控制器MyErrorController，并在Spring Boot主类中配置了错误路径。

以上就是Shiro与Spring Security用自定义异常处理401错误的攻略。这里提供两个示例，你可以根据自己的需要选择其中一种进行实现。