公司的API接口被刷了，那是因为你没这样做

点击上方“方志朋”，选择“设为星标”

回复”666“获取新整理的面试资料

作者：海向
来源： cnblogs.com/haixiang/p/12012728.html

api限流的场景

限流的需求出现在许多常见的场景中

1.秒杀活动，有人使用软件恶意刷单抢货，需要限流防止机器参与活动

2.某api被各式各样系统广泛调用，严重消耗网络、内存等资源，需要合理限流

3.淘宝获取ip所在城市接口、微信公众号识别微信用户等开发接口，免费提供给用户时需要限流，更具有实时性和准确性的接口需要付费。

api限流实战

首先我们编写注解类AccessLimit，使用注解方式在方法上限流更优雅更方便！三个参数分别代表有效时间、最大访问次数、是否需要登录，可以理解为 seconds 内最多访问 maxCount 次。

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface AccessLimit {int seconds();int maxCount();boolean needLogin() default true;
}

限流的思路

1.通过路径:ip的作为key，访问次数为value的方式对某一用户的某一请求进行唯一标识

2.每次访问的时候判断key是否存在，是否count超过了限制的访问次数

3.若访问超出限制，则应response返回msg:请求过于频繁给前端予以展示

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;@Component
public class AccessLimtInterceptor implements HandlerInterceptor {@Autowiredprivate RedisService redisService;@Overridepublic boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {if (handler instanceof HandlerMethod) {HandlerMethod hm = (HandlerMethod) handler;AccessLimit accessLimit = hm.getMethodAnnotation(AccessLimit.class);if (null == accessLimit) {return true;}int seconds = accessLimit.seconds();int maxCount = accessLimit.maxCount();boolean needLogin = accessLimit.needLogin();if (needLogin) {//判断是否登录}String key = request.getContextPath() + ":" + request.getServletPath() + ":" + ip ;Integer count = redisService.get(key);if (null == count || -1 == count) {redisService.set(key, 1);redisService.expire(seconds);return true;}if (count < maxCount) {redisService.inCr(key);return true;}if (count >= maxCount) {
// response 返回 json 请求过于频繁请稍后再试return false;}}return true;}
}

注册拦截器并配置拦截路径和不拦截路径

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;// extends WebMvcConfigurerAdapter 已经废弃，java 8开始直接继承就可以
@Configuration
public class IntercepterConfig  implements WebMvcConfigurer {@Autowiredprivate AccessLimtInterceptor accessLimtInterceptor;@Overridepublic void addInterceptors(InterceptorRegistry registry) {registry.addInterceptor(accessLimtInterceptor).addPathPatterns("/拦截路径").excludePathPatterns("/不被拦截路径 通常为登录注册或者首页");}
}

在Controller层的方法上直接可以使用注解@AccessLimit

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;@RestController
@RequestMapping("test")
public class TestControler {@GetMapping("accessLimit")@AccessLimit(seconds = 3, maxCount = 10)public String testAccessLimit() {//xxxxreturn "";}
}

说句题外话，springboot全家桶技术交流群可以加我微信，但是坑位有限哦，由于忙于工作，有时不能及时回复大家，请多包涵。

热门内容：
Mybatis：颠覆你心中对事务的理解为什么强烈推荐 Java 程序员使用 Google Guava 编程！IntelliJ IDEA 快捷键终极大全，速度收藏！「Jenkins+Git+Maven+Shell+Tomcat持续集成」经典教程  线上服务 CPU 又 100% 啦？一键定位 so easy！
GitHub 上有个沙雕开发者，做了款斗图工具后火了...大白话带你认识Kafka
如何参与一个顶级开源项目    深入理解 Spring Cloud 核心组件与底层原理最近面试BAT，整理一份面试资料《Java面试BAT通关手册》，覆盖了Java核心技术、JVM、Java并发、SSM、微服务、数据库、数据结构等等。
获取方式：点“在看”，关注公众号并回复 666 领取，更多内容陆续奉上。
明天见(｡･ω･｡)ﾉ♡