springSecurity之如何添加自定义过滤器

下面是关于“如何添加自定义过滤器到springSecurity中”的完整攻略：

添加自定义过滤器

在使用springSecurity时，有时候需要添加自定义的过滤器来实现一些特定的需求。下面我们就来介绍如何添加自定义的过滤器。

1. 定义自定义过滤器类

首先我们需要定义一个自定义过滤器类，这个过滤器类需要继承OncePerRequestFilter类，并实现doFilterInternal方法来进行过滤逻辑的处理，示例如下：

public class CustomFilter extends OncePerRequestFilter {

    @Override
    protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
        // 过滤逻辑处理
        filterChain.doFilter(request, response);
    }

}

1. 配置自定义过滤器

接下来需要在SpringSecurity配置类中配置自定义过滤器，使用http.addFilterBefore或http.addFilterAfter方法来添加到过滤器链中。示例如下：

@Configuration
@EnableWebSecurity
public class SecurityConfiguration extends WebSecurityConfigurerAdapter {

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        // 配置http相关信息
        http.addFilterBefore(new CustomFilter(), BasicAuthenticationFilter.class);
    }

}

这里使用的是addFilterBefore方法，在BasicAuthenticationFilter前添加自定义过滤器。

1. 配置过滤器名称

最后需要配置自定义过滤器的名称，可以通过@Component注解来为CustomFilter类配置名称。

@Component("customFilter")
public class CustomFilter extends OncePerRequestFilter {
    ...
}

接下来我们来看两个示例：

示例1：自定义一个IP过滤器

现在我们需要实现一个IP过滤器，只允许指定IP的用户访问我们的应用，其他用户都将被禁止。

1. 自定义IPFilter

我们定义一个IPFilter类来实现IP过滤器，这里我们从请求中获取用户的IP地址，判断其是否包含在指定的IP列表中，如果包含则允许访问，否则禁止访问。

@Component("ipFilter")
public class IPFilter extends OncePerRequestFilter {

    private final List<String> ipList = Arrays.asList("127.0.0.1", "localhost");

    @Override
    protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
        String remoteAddr = request.getRemoteAddr();
        if (ipList.contains(remoteAddr)) {
            filterChain.doFilter(request, response);
        } else {
            response.getWriter().write("Access Denied");
        }
    }
}

1. 配置IPFilter

接着我们需要在SpringSecurity配置类中配置IPFilter，将其添加到过滤器链中，这里使用的是addFilterBefore方法，在BasicAuthenticationFilter前添加自定义过滤器。

@Configuration
@EnableWebSecurity
public class SecurityConfiguration extends WebSecurityConfigurerAdapter {

    @Autowired
    private IPFilter ipFilter;

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http
                ...
                .addFilterBefore(ipFilter, BasicAuthenticationFilter.class);
    }
}

这样就完成了一个IP过滤器的实现和配置。

示例2：实现Token认证过滤器

现在我们需要实现一个Token认证过滤器，用来对用户的Token进行认证，如果验证失败，则禁止用户访问我们的应用。

1. 自定义TokenAuthFilter

我们自定义一个TokenAuthFilter类来实现Token认证过滤器，这里我们从请求头中获取Token值，并对其进行认证，如果认证成功，则允许访问，否则禁止访问。

@Component("tokenAuthFilter")
public class TokenAuthFilter extends OncePerRequestFilter {

    @Override
    protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
        String token = request.getHeader("Authorization");
        if ("123456".equals(token)) {
            filterChain.doFilter(request, response);
        } else {
            response.getWriter().write("Access Denied");
        }
    }
}

1. 配置TokenAuthFilter

接着我们需要在SpringSecurity配置类中配置TokenAuthFilter，将其添加到过滤器链中，这里使用的是addFilterBefore方法，在BasicAuthenticationFilter前添加自定义过滤器。

@Configuration
@EnableWebSecurity
public class SecurityConfiguration extends WebSecurityConfigurerAdapter {

    @Autowired
    private TokenAuthFilter tokenAuthFilter;

    @Override
    protected void configure(HttpSecurity http) throws Exception {
        http
                ...
                .addFilterBefore(tokenAuthFilter, BasicAuthenticationFilter.class);
    }
}

这样就完成了一个Token认证过滤器的实现和配置。