基于spring-security 401 403錯誤自定義處理方案

更新時間：2021年07月26日 15:41:30 作者：PolarisHuster

這篇文章主要介紹了基于spring-security 401 403錯誤自定義處理方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

spring-security 401 403錯誤自定義處理

為了返回給前端統(tǒng)一的數(shù)據(jù)格式，

一般所有的數(shù)據(jù)都會以類似下面的方式返回：

public class APIResultDto<T> {
    /**
     * 狀態(tài)碼：-1代表成功,具體參考APIErrorCode類
     */
    private int er;
 
    /**
     * 狀態(tài)描述，可以自行設置或使用APIErrorCode類中默認描述
     */
    private String erMessage;
 
    /**
     * 實際返回實體，isSuccess()返回true時該字段有效
     */
    private T items;
}

但是一些框架，比如本文要說的spring-security是不按照我們自定義規(guī)范處理的，幸運的是spring-security框架給了我們可以定制化的地方，只需繼承

ResourceServerConfigurerAdapter

重寫

public void configure(ResourceServerSecurityConfigurer resources) throws Exception

方法即可，在里面添加自定義的針對授權(quán)時返回的401以及403錯誤碼，

具體如下：

@Autowired
    private AccessDeniedHandler accessDeniedHandler;
    @Autowired
    private AuthenticationEntryPoint authenticationEntryPoint;
 
    @Override
    public void configure(ResourceServerSecurityConfigurer resources) {
        resources.authenticationEntryPoint(authenticationEntryPoint);
        resources.accessDeniedHandler(accessDeniedHandler);
    }

里面涉及到的AccessDeniedHandler以及AuthenticationEntryPoint

如下所示：

@Component
public class CustomizedAuthenticationEntryPoint implements AuthenticationEntryPoint {
 
    @Override
    public void commence(HttpServletRequest request, HttpServletResponse response,
                         AuthenticationException authException) throws IOException, ServletException {
        response.setContentType("application/json;charset=UTF-8");
        
         //按照系統(tǒng)自定義結(jié)構(gòu)返回授權(quán)失敗
response.getWriter().write(JSON.toJSONString(APIResultDto.failed(APIErrorCode.AUTH_FAILED)));
    }
}
@Component
public class CustomAccessDeniedHandler implements AccessDeniedHandler {
 
    @Override
    public void handle(HttpServletRequest request, HttpServletResponse response, AccessDeniedException accessDeniedException) throws IOException, ServletException {
        response.setContentType("application/json;charset=UTF-8");
        
          //按照系統(tǒng)自定義結(jié)構(gòu)返回授權(quán)失敗
 response.getWriter().write(JSON.toJSONString(APIResultDto.failed(APIErrorCode.AUTH_FAILED)));
    }
}