Spring boot拦截器实现IP黑名单的完整步骤

 更新时间:2020年06月14日 09:04:58   作者:烟雨先生  
这篇文章主要给大家介绍了关于Spring boot拦截器实现IP黑名单的完整步骤,文中通过示例代码介绍的非常详细,对大家学习或者使用Spring boot拦截器具有一定的参考学习价值,需要的朋友们下面来一起学习学习吧

一·业务场景和需要实现的功能

以redis作为IP存储地址实现。

业务场景:针对秒杀活动或者常规电商业务场景等,防止恶意脚本不停的刷接口。

实现功能:写一个拦截器拦截掉黑名单IP,额外增加一个接口,将ip地址添加到redis中,并且返回redis中当前全部ip

二·Springboot中定义一个拦截器

@Order(0)
@Aspect
@Component
public class AopInterceptor {


 /**
  * 定义拦截器规则
  */
 @Pointcut("execution(* com.test.test.api.controller.test.test.*(..))")
 public void pointCut() {
 }

  /**
  * 拦截器具体实现
  *
  * @throws Throwable
  */
 @Around(value = "pointCut()")
 public Object around(ProceedingJoinPoint point) throws Throwable {
  try {

   HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
   //判断是否为黑名单用户
   String ip = getIpAddress(request);
   if (checkIpBlack(ip)) {
    //ip在黑名单中返回false
    //return false;
    DefaultResponse defaultResponse = new DefaultResponse();
    defaultResponse.setCode(-1);
    defaultResponse.setMessage("ip在黑名单中,拒绝访问.");
    SysLogHelper.log("IpBlackAopInterceptor", "当前请求ip" + ip, "ip在黑名单中,拒绝访问");
    return defaultResponse;
   } else {
    //ip不在黑名单中返回true
    SysLogHelper.log("IpBlackAopInterceptor", "当前请求ip" + ip, "ip正常,允许访问");
    return point.proceed();
   }


  } catch (Exception e) {
   e.printStackTrace();
   SysLogHelper.error("IpBlackAopInterceptor黑名单拦截异常:", ExceptionUtils.getMessage(e) + "详细" + ExceptionUtils.getStackTrace(e), null);
  }
  return point.getArgs();
 }

 //对比当前请求IP是否在黑名单中,注意(对比黑名单ip存放在redis中)
 public boolean checkIpBlack(String ip) throws Exception {
  IpBlackBody body = new IpBlackBody();
  body = cacheHelper.get("IpBlack:ips", IpBlackBody.class);
  if (body != null) {
   for (int i = 0; i < body.getIp().length; i++) {
    if (body.getIp()[i].equals(ip))
     return true;
   }
  }
  return false;
 }

}

三·获取请求主机IP地址

 public final static String getIpAddress(HttpServletRequest request)
   throws IOException {
  // 获取请求主机IP地址,如果通过代理进来,则透过防火墙获取真实IP地址

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

  if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
   if (ip == null || ip.length() == 0
     || "unknown".equalsIgnoreCase(ip)) {
    ip = request.getHeader("Proxy-Client-IP");
   }
   if (ip == null || ip.length() == 0
     || "unknown".equalsIgnoreCase(ip)) {
    ip = request.getHeader("WL-Proxy-Client-IP");
   }
   if (ip == null || ip.length() == 0
     || "unknown".equalsIgnoreCase(ip)) {
    ip = request.getHeader("HTTP_CLIENT_IP");
   }
   if (ip == null || ip.length() == 0
     || "unknown".equalsIgnoreCase(ip)) {
    ip = request.getHeader("HTTP_X_FORWARDED_FOR");
   }
   if (ip == null || ip.length() == 0
     || "unknown".equalsIgnoreCase(ip)) {
    ip = request.getRemoteAddr();
   }
  } else if (ip.length() > 15) {
   String[] ips = ip.split(",");
   for (int index = 0; index < ips.length; index++) {
    String strIp = (String) ips[index];
    if (!("unknown".equalsIgnoreCase(strIp))) {
     ip = strIp;
     break;
    }
   }
  }

  return ip;
 }

四·扩展接口,实现将黑名单IP写入redis当中,并返回当前所有黑名单IP

@RestController
public class IpBlackController {

 @Autowired(required = false)
 private CacheHelper cacheHelper;

 @PostMapping("/testIpBlack")
 public IpBlackBody IpBlack(@RequestBody IpBlackBody ipBlackBody) throws Exception {

  IpBlackBody body = new IpBlackBody();
  body = cacheHelper.get("IpBlack:ips", IpBlackBody.class);

  if (body != null) {
   //拼接当前IP与redis中现有ip
   linkArray(body.getIp(), ipBlackBody.getIp());
   //将数据赋给body
   body.setIp(linkArray(body.getIp(), ipBlackBody.getIp()));
   //setex中第二个参数时间为S,根据业务场景相应调整,此处我设置为一天
   //将body中拼接后的ip地址数据写入redis中
   cacheHelper.setex("IpBlack:ips", 86400, body);

  } else {
   cacheHelper.setex("IpBlack:ips", 86400, ipBlackBody);
   body = cacheHelper.get("IpBlack:ips", IpBlackBody.class);
   return body;
  }
  return body;
 }

 //拼接两个String[]的方法
 public static String[] linkArray(String[] array1, String[] array2) {

  List<String> list = new ArrayList<>();
  if (array1 == null) {
   return array2;
  }
  if (array2 == null) {
   return array1;
  }
  for (int i = 0; i < array1.length; i++) {
   list.add(array1[i]);
  }
  for (int i = 0; i < array2.length; i++) {
   list.add(array2[i]);
  }
  String[] returnValue = new String[list.size()];
  for (int i = 0; i < list.size(); i++) {

   returnValue[i] = list.get(i);
  }
  return returnValue;
 }
}

总结:

首先根据需要拦截的controller拦截响应请求controller层,然后根据编写相关拦截器的具体实现,其中包含两部主要操作:

1.获取到远程请求主机的实际ip地址

2.对比当前ip是否在黑名单中(此次操作需要读取redis中的黑名单ip列表)

然后根据当前需求增加了一个redis接口,实现将需要封禁的IP地址增加到redis黑名单中并返回当前所有的黑名单IP地址。

至此:至此springboot通过拦截器实现拦截黑名单功能已经实现。

到此这篇关于Spring boot拦截器实现IP黑名单的文章就介绍到这了,更多相关Springboot拦截器IP黑名单内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Java查看本机端口是否被占用源码

    Java查看本机端口是否被占用源码

    这篇文章主要介绍了Java查看本机端口是否被占用的主要原理,并结合具体实例给出了操作方法,需要的朋友可以参考下
    2017-09-09
  • 详解Java中的封装、继承、多态

    详解Java中的封装、继承、多态

    本文主要介绍了Java中的封装、继承、多态的相关知识,具有一定的参考价值,下面跟着小编一起来看下吧
    2017-01-01
  • Java客户端利用Jedis操作redis缓存示例代码

    Java客户端利用Jedis操作redis缓存示例代码

    Jedis是Redis官方推荐的用于访问Java客户端,下面这篇文章主要给大家介绍了关于Java客户端利用Jedis操作redis缓存的相关资料,文中给出了详细的示例代码,需要的朋友可以参考借鉴,下面来一起看看吧。
    2017-07-07
  • java使用Base64实现文件加密解密

    java使用Base64实现文件加密解密

    这篇文章主要为大家详细介绍了java实现Base64给文件加密、解密,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-03-03
  • Java Reactor反应器模式使用方法详解

    Java Reactor反应器模式使用方法详解

    这篇文章主要介绍了Java Reactor反应器模式使用方法详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-06-06
  • SpringBoot 日志的配置及输出应用教程

    SpringBoot 日志的配置及输出应用教程

    Spring Boot 默认使用 SLF4J+Logback 记录日志,并提供了默认配置。本文我们将重点介绍Spring Boot日志的配置及输出。感兴趣的小伙伴可以了解一下
    2021-12-12
  • JAVA项目常用异常处理汇总

    JAVA项目常用异常处理汇总

    这篇文章主要介绍了JAVA项目常用异常处理汇总,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-11-11
  • Java中的单向链表详解

    Java中的单向链表详解

    这篇文章主要介绍了Java中的单向链表详解,单向链表又叫单链表,是链表的一种,由节点构成,head指针指向第一个称为表头节点,而终止指向最后一个null指针,需要的朋友可以参考下
    2024-01-01
  • Spring 多线程事务控制的实践

    Spring 多线程事务控制的实践

    本文主要介绍了Spring 多线程事务控制的实践,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2023-09-09
  • Java中toString()、String.valueOf、(String)强转区别

    Java中toString()、String.valueOf、(String)强转区别

    相信大家在日常开发中这三种方法用到的应该很多,本文主要介绍了Java中toString()、String.valueOf、(String)强转区别,感兴趣的可以了解一下
    2021-09-09

最新评论