Spring Boot(五十五):基于redis防止接口恶意刷新和暴力请求
创始人
2024-05-13 08:19:22
0

下面的教程,通过intercept和redis针对url+ip在一定时间内访问的次数来将ip禁用,可以根据自己的需求进行相应的修改,来达到自己的目的

下面只讲解大致步骤,不详细讲解,需要完整代码的可以自行下载。

https://download.csdn.net/download/u013938578/87389620

新建SpringBoot项目:

 拦截器类如下:

        对访问ip地址进行识别和判断。

package com.example.demo.Interceptor;import com.alibaba.fastjson.JSON;
import com.example.demo.Utils.IpAdrressUtil;
import com.example.demo.Utils.RedisUtil;
import com.example.demo.entity.Result;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;@Slf4j
@Component
@RequiredArgsConstructor
public class IpUrlLimitInterceptor implements HandlerInterceptor {private final RedisUtil redisUtil;private static final String LOCK_IP_URL_KEY = "lock_ip_";private static final String IP_URL_REQ_TIME = "ip_url_times_";private static final long LIMIT_TIMES = 5;private static final String IP_LOCK_TIME = "60";@Overridepublic boolean preHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o) throws Exception {log.info("request请求地址uri={},ip={}", httpServletRequest.getRequestURI(), IpAdrressUtil.getIpAddr(httpServletRequest));if (ipIsLock(IpAdrressUtil.getIpAddr(httpServletRequest))) {log.info("ip访问被禁止={}", IpAdrressUtil.getIpAddr(httpServletRequest));returnJson(httpServletResponse, JSON.toJSONString(Result.success("ip访问被禁止")));return false;}if (!addRequestTime(IpAdrressUtil.getIpAddr(httpServletRequest), httpServletRequest.getRequestURI())) {returnJson(httpServletResponse, JSON.toJSONString(Result.success("ip访问被禁止")));return false;}return true;}@Overridepublic void postHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, ModelAndView modelAndView) throws Exception {}@Overridepublic void afterCompletion(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, Exception e) throws Exception {}/*** 判断ip是否被禁用** @param ip* @return*/private Boolean ipIsLock(String ip) {if (redisUtil.hasKey(LOCK_IP_URL_KEY + ip)) {return true;}return false;}/*** 记录请求次数** @param ip* @param uri* @return*/private Boolean addRequestTime(String ip, String uri) {String key = IP_URL_REQ_TIME + ip + uri;if (redisUtil.hasKey(key)) {// 如果key存在,次数+1long time = redisUtil.incr(key, (long) 1);log.info("time:{}", time);if (time >= LIMIT_TIMES) {// 如果超过限制次数,则设置ip被禁用 60秒redisUtil.getLock(LOCK_IP_URL_KEY + ip, ip, IP_LOCK_TIME);return false;}} else {// ip+uri请求次数为1,1秒后过期redisUtil.getLock(key, "1", "1");log.info("记录请求次数1");}return true;}private void returnJson(HttpServletResponse response, String json) throws Exception {PrintWriter writer = null;response.setCharacterEncoding("UTF-8");response.setContentType("text/json; charset=utf-8");try {writer = response.getWriter();writer.print(json);} catch (IOException e) {log.error("LoginInterceptor response error ---> {}", e.getMessage(), e);} finally {if (writer != null) {writer.close();}}}}

拦截器配置文件:

package com.example.demo.Config;import com.example.demo.Interceptor.IpUrlLimitInterceptor;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;@Configuration
@Slf4j
@RequiredArgsConstructor
public class MyWebAppConfig implements WebMvcConfigurer {private final IpUrlLimitInterceptor ipUrlLimitInterceptor;@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(ipUrlLimitInterceptor).addPathPatterns("/**");}
}

代码中redis的使用的是分布式锁的形式,这样可以最大程度保证线程安全和功能的实现效果。代码中设置的是1S内同一个接口通过同一个ip访问5次,就将该ip禁用1个小时,根据自己项目需求可以自己适当修改,实现自己想要的功能;

redis分布式锁代码:

package com.example.demo.Utils;import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.core.script.RedisScript;
import org.springframework.stereotype.Component;import java.util.Collections;@Component
@Slf4j
@RequiredArgsConstructor
public class RedisUtil {private static final Long SUCCESS = 1L;private final RedisTemplate redisTemplate;/*** 获取锁** @param lockKey* @param value* @param expireTime:单位-秒* @return*/public boolean getLock(String lockKey, Object value, String expireTime) {boolean ret = false;try {String script = "if redis.call('setNx',KEYS[1],ARGV[1]) then if redis.call('get',KEYS[1])==ARGV[1] then return redis.call('expire',KEYS[1],ARGV[2]) else return 0 end end";RedisScript redisScript = new DefaultRedisScript<>(script, Long.class);Long result = redisTemplate.execute(redisScript, Collections.singletonList(lockKey), value, expireTime);log.info("result:{}", result);if (SUCCESS.equals(result)) {return true;}} catch (Exception e) {log.error("getLock error:{}", e.getMessage(), e);}return ret;}/*** 释放锁** @param lockKey* @param value* @return*/public boolean releaseLock(String lockKey, String value) {String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";RedisScript redisScript = new DefaultRedisScript<>(script, String.class);Object result = redisTemplate.execute(redisScript, Collections.singletonList(lockKey), value);if (SUCCESS.equals(result)) {return true;}return false;}/*** 设置一个自增的数据** @param key* @param num*/public Long incr(String key, Long num) {return redisTemplate.opsForValue().increment(key, num);}public Boolean hasKey(String key) {return redisTemplate.hasKey(key);}}

获取Ip地址功能类如下:

package com.example.demo.Utils;import javax.servlet.http.HttpServletRequest;
import java.net.InetAddress;
import java.net.UnknownHostException;public class IpAdrressUtil {/*** 获取IP地址*/public static String getIpAddr(HttpServletRequest request){String 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") || ipAddress.equals("0:0:0:0:0:0:0:1")){//根据网卡取本机配置的IPInetAddress inet=null;try {inet = InetAddress.getLocalHost();} catch (UnknownHostException e) {}if (inet.getHostAddress() != null) {ipAddress= inet.getHostAddress();}}}//对于通过多个代理的情况,第一个IP为客户端真实IP,多个IP按照','分割if(ipAddress!=null && ipAddress.length()>15){ //"***.***.***.***".length() = 15if(ipAddress.indexOf(",")>0){ipAddress = ipAddress.substring(0,ipAddress.indexOf(","));}}return ipAddress;}
}

第一次访问接口结果如下:

当快速多次访问接口以后,接口返回结果如下:

查看redis发现已经对ip地址加上了锁:

相关内容

热门资讯

骆驼祥子的读书笔记 关于骆驼祥子的读书笔记15篇  当看完一本著作后,大家一定对生活有了新的感悟和看法,不妨坐下来好好写...
《答谢中书书》作者陶弘景的生... 《答谢中书书》作者陶弘景的生平简介  引导语:《答谢中书书》是南朝文学家陶弘景写给朋友的一封书信。文...
优美段落摘抄及点评 优美段落摘抄及点评  段落,汉语词语,拼音是,意思是指根据文章或事情的内容、阶段划分的相对独立的部分...
《红楼梦》人名谜语及答案 《红楼梦》人名谜语大全及答案  猜谜语是一种很有趣的.益智游戏。谜面总是简洁形象地把谜底描述出来。以...
岳阳楼记说课稿 岳阳楼记说课稿8篇  作为一位无私奉献的人民教师,就不得不需要编写说课稿,说课稿是进行说课准备的文稿...
老马识途文言文阅读 老马识途文言文阅读  文言文阅读  老马识途  管仲、隰朋从桓公伐孤竹,春往冬反,迷惑失道。管仲曰:...
《师说》文言文原文及赏析 《师说》文言文原文及赏析  《师说》作者是唐朝文学家韩愈。其全文如下:  【前言】  《师说》作于唐...
跨文化视角下西方文学作品的鉴... 跨文化视角下西方文学作品的鉴赏和语文翻译  【摘 要】基于跨文化的视角,简要地对西方文学作品进行了赏...
《易经》第十卦·履卦 《易经》第十卦·履卦  《周易》,又称《易经》,简称《易》。易经分两部分,一为“经”为伏羲氏和周文王...
《红楼梦》的十二钗 《红楼梦》的十二钗  梦阮先生潦居悼红轩作《红楼梦》,佳作未成身先葬,可怜侯门梦难成。  初读《红楼...
韩愈《劝学篇》全文 韩愈《劝学篇》全文  《进学解》是元和七、八年间韩愈任国子博士时所作,是韩愈所创作的一篇关于劝学的文...
李鳝《蕉石图》鉴赏 李鳝《蕉石图》鉴赏  李鳝出身富豪之家,早年曾以画人内廷供奉,后来则转任山东滕县知县。李鳝画《蕉石图...
张俭传文言文阅读及答案 张俭传文言文阅读及答案  张俭字元节,山阳高平人,赵王张耳之后也,父成,江夏太守。俭初举茂才,以刺史...
心经全文注音朗诵 心经全文注音朗诵  导语:凡人要度苦厄,了生死,成大觉,非从自心下手不可。以下小编为大家介绍心经全文...
范元琰良善文言文翻译 范元琰良善文言文翻译  范元琰,字伯珪,吴郡钱塘人也。小编整理的范元琰良善文言文翻译,欢迎大家前来查...
韩非子权术全文译文 韩非子权术全文译文  引导语:《韩非子权术》这篇课文想必很多人都读过,那么相关的韩非子权术的全文以及...
《不食嗟来之食》原文注释及翻... 《不食嗟来之食》原文注释及翻译  《不食嗟来之食》,选自《礼记。檀弓》。《礼记》,是中国古代一部重要...
青门引 张先 赏析 青门引 张先 赏析 千秋岁 数声鶗鴂 张先,是作者写的一篇伤春词,同时抒发自己寥落寂寞的处境。本文由...
苏东坡红梅原文及翻译 苏东坡红梅原文及翻译  宋神宗元丰五年(1082),当时苏轼贬官在黄州,因读石延年《红梅》诗引起感触...
父亲节英文祝福语 父亲节英文祝福语(精选50句)  有一种深情叫做父子连心,有一种真爱叫做父爱如山。父亲的爱庄重而深刻...