👉 这是一个或许对你有用的社群
🐱 一对一交流/面试小册/简历优化/求职解惑，欢迎加入「芋道快速开发平台」知识星球。下面是星球提供的部分资料：
👉这是一个或许对你有用的开源项目
国产 Star 破 10w+ 的开源项目，前端包括管理后台 + 微信小程序，后端支持单体和微服务架构。
功能涵盖 RBAC 权限、SaaS 多租户、数据权限、商城、支付、工作流、大屏报表、微信公众号、CRM 等等功能：
  • Boot 仓库：https://gitee.com/zhijiantianya/ruoyi-vue-pro
  • Cloud 仓库：https://gitee.com/zhijiantianya/yudao-cloud
  • 视频教程：https://doc.iocoder.cn
【国内首批】支持 JDK 21 + SpringBoot 3.2.2、JDK 8 + Spring Boot 2.7.18 双版本 
来源：blog.csdn.net/second_place_zyj

/article/details/130320957
实现流程： 自定义拦截器——注入拦截器——获取请求IP——对比IP是否一致——请求返回
文章背景： 接口添加IP白名单限制，只有规定的IP可以访问项目。
实现思路： 添加拦截器，拦截项目所有的请求，获取请求的网络IP，查询IP是否在白名单之中，白名单设置在数据库中，用一张表存储，若在表中有此IP则进行下一步，不在则进行请求拦截，返回到客户端。
实现方式： HandlerInterceptor+MySQL+Mybatis-plus
自定义拦截器，创建类并且实现HandlerInterceptor接口，即可成为拦截器。
HandlerInterceptor接口提供了三个方法，三个方法分别如下
HandlerInterceptor接口提供了三个方法，三个方法分别如下
preHandle方法在业务处理器处理请求之前进行调用，可以用做权限校验，token验证等等。结果返回true继续执行，结果返回false拦截请求。
postHandle方法在业务处理器处理请求执行完成后，生成视图之前执行。
afterCompletion方法在业务处理器执行完成之后，可以用做清理资源，日志记录等等。
自定义拦截器：实现HandlerInterceptor接口，重写preHandle方法，在preHandle添加获取IP的方法和IP检验业务。代码中涉及到的testEngineerService.getIp()方法在下边！！！
import com.alibaba.fastjson.JSON;

import com.lifel.service.TestEngineer.TestEngineerService;

import com.lifel.utils.ToolsResultEntity;

import lombok.extern.slf4j.Slf4j;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.web.context.WebApplicationContext;

import org.springframework.web.context.support.WebApplicationContextUtils;

import org.springframework.web.servlet.HandlerInterceptor;


import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import java.io.IOException;

import java.io.PrintWriter;

import java.net.InetAddress;

import java.net.UnknownHostException;


/********************************************************************************

 ** @author ：ZYJ

 ** @date ：2023/04/23

 ** @description ：自定义拦截器 拦截ip

 *********************************************************************************/

@Slf4j

public class WhiteListIntercept implements HandlerInterceptor {


    @Autowired

    private TestEngineerService testEngineerService;


    @Override

    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

        String ipAddress = null;

        try {

            ipAddress = request.getHeader(
"x-forwarded-for"
);

if
 (ipAddress == null || ipAddress.length() == 0 || 
"unknown"
.equalsIgnoreCase(ipAddress)) {

                ipAddress = request.getHeader(
"Proxy-Client-IP"
);

            }

if
 (ipAddress == null || ipAddress.length() == 0 || 
"unknown"
.equalsIgnoreCase(ipAddress)) {

                ipAddress = request.getHeader(
"WL-Proxy-Client-IP"
);

            }

if
 (ipAddress == null || ipAddress.length() == 0 || 
"unknown"
.equalsIgnoreCase(ipAddress)) {

                ipAddress = request.getRemoteAddr();

if
 (ipAddress.equals(
"127.0.0.1"
)) {

                    // 根据网卡取本机配置的IP

                    InetAddress inet = null;

                    try {

                        inet = InetAddress.getLocalHost();

                    } catch (UnknownHostException e) {

                        e.printStackTrace();

                    }

                    ipAddress = inet.getHostAddress();

                }

            }

            // 对于通过多个代理的情况，第一个IP为客户端真实IP,多个IP按照
','
分割

if
 (ipAddress != null && ipAddress.length() > 15) {

if
 (ipAddress.indexOf(
","
) > 0) {

                    ipAddress = ipAddress.substring(0, ipAddress.indexOf(
","
));

                }

            }

        } catch (Exception e) {

            ipAddress=
""
;

        }

        log.info(
"机主的ip是"
+ipAddress);

        WebApplicationContext cxt = WebApplicationContextUtils.getWebApplicationContext(request.getServletContext());

if
(cxt != null && cxt.getBean(TestEngineerService.class) != null &&testEngineerService == null) {

            testEngineerService =cxt.getBean(TestEngineerService.class);

        }

if
(testEngineerService.getIp(ipAddress)){

returntrue
;

        }
else
{

            returnJson(response, JSON.toJSONString(new ToolsResultEntity(1002, 
"ip不存在"
, null)));

returnfalse
;

        }

    }


    /********************************************************************************

     ** @author ：ZYJ

     ** @date ：2023/04/23

     ** @description ：设置请求拦截返回参数

     *********************************************************************************/

    private void returnJson(HttpServletResponse response, String result) {

        PrintWriter writer = null;

        response.setCharacterEncoding(
"UTF-8"
);

        response.setContentType(
"text/html; charset=utf-8"
);

        try {

            writer = response.getWriter();

            writer.print(result);

        } catch (IOException e) {

        } finally {

if
 (writer != null) {

                writer.close();

            }

        }

    }

}

注入拦截器：将拦截器注入到spring，交给spring管理
import org.springframework.context.annotation.Configuration;

import org.springframework.web.servlet.config.annotation.InterceptorRegistry;

import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;


@Configuration

public class AdminWebConfig implements WebMvcConfigurer {


    /********************************************************************************

     ** @author ：ZYJ

     ** @date ：2023/04/23

     ** @description ：配置拦截器

     *********************************************************************************/

    @Override

    public void addInterceptors(InterceptorRegistry registry) {

//       下面这句代码相当于添加一个拦截器，添加的拦截器就是我们刚刚创建的

        registry.addInterceptor(new WhiteListIntercept())

//       addPathPatterns()配置我们要拦截哪些路径 addPathPatterns(
"/**"
)表示拦截所有请求，包括我们的静态资源

                .addPathPatterns(
"/**"
);

    }

}

创建Service和实现类
public interface TestEngineerService {


    /********************************************************************************

     ** @author ：ZYJ

     ** @date ：2023/04/23

     ** @description ：查询IP是否在白名单中

     *********************************************************************************/

    Boolean getIp(String name);


}

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;

import com.lifel.entity.TestEngineer.WhiteIp;

import com.lifel.mapper.TestEngineer.WhiteIpMapper;

import com.lifel.service.TestEngineer.TestEngineerService;

import org.springframework.stereotype.Service;


import javax.annotation.Resource;


@Service

public class TestEngineerServiceImpl implements TestEngineerService {


    @Resource

    private WhiteIpMapper whiteIpMapper;


    /********************************************************************************

     ** @author ：ZYJ

     ** @date ：2023/04/23

     ** @description ：查询IP是否在数据库中保存

     *********************************************************************************/

    @Override

    public Boolean getIp(String name) {

        try {

            //查询表中是否有此IP，并且状态是打开的状态

            WhiteIp whiteIp = whiteIpMapper.selectOne(new QueryWrapper<WhiteIp>().eq(
"name"
, name).eq(
"state"
, 
"open"
));

if
(whiteIp==null){

returnfalse
;

            }
else
{

returntrue
;

            }

        }catch (Exception e){

returnfalse
;

        }

    }

}

测试代码，创建Controller，调用方法进行测试
import io.swagger.annotations.Api;

import io.swagger.annotations.ApiOperation;

import org.springframework.web.bind.annotation.PostMapping;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;


@RestController

@RequestMapping(value = 
"/api"
)

@Api(value = 
"测试类"
, tags = 
"测试类"
)

public class TestEngineerController {


    @GetMapping(
"/test"
)

    @ApiOperation(value = 
"测试IP白名单是否生效"
)

    public String 
test
() {

return"ok"
;

    }

}

查看我们的IP地址，保存到数据库中，以保证只有我们自己的IP才能访问项目接口
表中保存我们正确的IP，启动项目访问测试方法，请求结果正常返回！！！
表中保存一个错误的IP，启动项目访问测试方法，请求拦截提示IP不存在！！！
“
注意：测试的时候不能写localhost，必须是127.0.0.1:8080，127.0.0.1获取的才是我们本地的IP，localhost获取的为0:0:0:0:0:0:0:1
欢迎加入我的知识星球，全面提升技术能力。
👉 加入方式，“长按”或“扫描”下方二维码噢：
星球的内容包括：项目实战、面试招聘、源码解析、学习路线。
文章有帮助的话，在看，转发吧。
谢谢支持哟 (*^__^*）
继续阅读
阅读原文