大橙子网站建设,新征程启航
为企业提供网站建设、域名注册、服务器等服务
小编给大家分享一下SpringMVC如何实现限流,希望大家阅读完这篇文章之后都有所收获,下面让我们一起去探讨吧!
目前创新互联已为成百上千家的企业提供了网站建设、域名、虚拟主机、绵阳服务器托管、企业网站设计、西峰网站维护等服务,公司将坚持客户导向、应用为本的策略,正道将秉承"和谐、参与、激情"的文化,与客户和合作伙伴齐心协力一起成长,共同发展。
使用说明
在项目中引入Guava相关包
http://mvnrepository.com/artifact/com.google.guava/guava/21.0
maven项目
com.google.guava guava 21.0
gradle项目
// https://mvnrepository.com/artifact/com.google.guava/guava compile group: 'com.google.guava', name: 'guava', version: '21.0'
写一个SpringMVC的拦截器
SmoothBurstyInterceptor.java
import com.google.common.util.concurrent.RateLimiter; import org.springframework.web.servlet.ModelAndView; import org.springframework.web.servlet.handler.HandlerInterceptorAdapter; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.util.concurrent.TimeUnit; public class SmoothBurstyInterceptor extends HandlerInterceptorAdapter { public enum LimitType { DROP,//丢弃 WAIT //等待 } /** * 限流器 */ private RateLimiter limiter; /** * 限流方式 */ private LimitType limitType = LimitType.DROP; public SmoothBurstyInterceptor() { this.limiter = RateLimiter.create(10); } /** * @param tps 限流量 (每秒处理量) * @param limitType 限流类型:等待/丢弃(达到限流量) */ public SmoothBurstyInterceptor(int tps, SmoothBurstyInterceptor.LimitType limitType) { this.limiter = RateLimiter.create(tps); this.limitType = limitType; } /** * @param permitsPerSecond 每秒新增的令牌数 * @param limitType 限流类型:等待/丢弃(达到限流量) */ public SmoothBurstyInterceptor(double permitsPerSecond, SmoothBurstyInterceptor.LimitType limitType) { this.limiter = RateLimiter.create(permitsPerSecond, 1000, TimeUnit.MILLISECONDS); this.limitType = limitType; } @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { if (limitType.equals(LimitType.DROP)) { if (limiter.tryAcquire()) { return super.preHandle(request, response, handler); } } else { limiter.acquire(); return super.preHandle(request, response, handler); } throw new Exception("网络异常!");//达到限流后,往页面提示的错误信息。 } @Override public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception { super.postHandle(request, response, handler, modelAndView); } @Override public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { super.afterCompletion(request, response, handler, ex); } public RateLimiter getLimiter() { return limiter; } public void setLimiter(RateLimiter limiter) { this.limiter = limiter; } }
SpringMVC拦截配置
WebConfig.java
@Component public class WebConfig extends WebMvcConfigurerAdapter { @Override public void addInterceptors(InterceptorRegistry registry) { // 多个拦截器组成一个拦截器链 registry.addInterceptor(new SmoothBurstyInterceptor(100, SmoothBurstyInterceptor.LimitType.DROP)).addPathPatterns("/**"); //限流可配置为SmoothBurstyInterceptor.LimitType.DROP丢弃请求或者SmoothBurstyInterceptor.LimitType.WAIT等待,100为每秒的速率 super.addInterceptors(registry); } }
看完了这篇文章,相信你对“SpringMVC如何实现限流”有了一定的了解,如果想了解更多相关知识,欢迎关注创新互联行业资讯频道,感谢各位的阅读!