在使用SpringBoot做接口訪問如何做接口的限流，這里我們可以使用google的Guava包來實現，當然我們也可以自己實現限流，Guava中的限流是久經考驗的我們沒必需重新再去寫一個，如果想了解限流原理的同學可以自己查閱一下相關的資料，本文不作過來說明噢。
使用說明
在項目中引入Guava相關包
http://mvnrepository.com/artifact/com.google.guava/guava/21.0
maven項目
|
1
2
3
4
5
6
|
<!-- https://mvnrepository.com/artifact/com.google.guava/guava --><dependency> <groupId>com.google.guava</groupId> <artifactId>guava</artifactId> <version>21.0</version></dependency> |
gradle項目
|
1
2
|
// https://mvnrepository.com/artifact/com.google.guava/guavacompile group: 'com.google.guava', name: 'guava', version: '21.0' |
寫一個SpringMVC的攔截器
SmoothBurstyInterceptor.java
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
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
|
1
2
3
4
5
6
7
8
9
10
11
12
|
@Componentpublic 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); }} |
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持服務器之家。
原文鏈接：http://blog.csdn.net/dounine/article/details/71439809
