ip:
  whitelist: IP1  #白名單
  blacklist: IP2 #黑名單

@Component
public class IpFilter implements Filter {

    //白名單IP列表
    private List<String> whitelist;
    //黑名單IP列表
    private List<String> blacklist;


    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
        //獲取上下文信息
        ServletContext context = filterConfig.getServletContext();
        WebApplicationContext ctx = WebApplicationContextUtils.getWebApplicationContext(context);
        Environment environment = ctx.getBean(Environment.class);
        //賦予初值
        whitelist = Arrays.stream(environment.getProperty("ip.whitelist").split(","))
                .map(String::trim)
                .collect(Collectors.toList());
        blacklist=Arrays.stream(environment.getProperty("ip.blacklist").split(","))
                .map(String::trim)
                .collect(Collectors.toList());
    }




    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        HttpServletRequest httpRequest = (HttpServletRequest) request;
        String ip = httpRequest.getRemoteAddr();

        if (whitelist.contains(ip)) {
            // 白名單IP，直接通過
            chain.doFilter(request, response);
        } else if (blacklist.contains(ip)) {
            // 黑名單IP，拒絕訪問
            HttpServletResponse httpResponse = (HttpServletResponse) response;
            httpResponse.sendError(HttpServletResponse.SC_FORBIDDEN, "IP is in blacklist");
        } else {
            // 非黑白名單IP，根據(jù)業(yè)務(wù)需求處理（例如：允許或拒絕）
            chain.doFilter(request, response);
        }
    }


    @Override
    public void destroy() {
        // 清理資源，如果有的話
    }
}

@Configuration
public class FilterConfig {
    @Bean
    public FilterRegistrationBean<IpFilter> ipFilterRegistration() {
        FilterRegistrationBean<IpFilter> registrationBean = new FilterRegistrationBean<>();
        registrationBean.setFilter(new IpFilter());
        registrationBean.addUrlPatterns("/*"); // 設(shè)置過濾器攔截的路徑
        registrationBean.setName("ipFilter");
        registrationBean.setOrder(1); // 設(shè)置過濾器的順序
        return registrationBean;
    }

}