Guava RateLimiter 实现 API 限流，这才是正确的姿势！

public static void main(String[] args) {

String start = new SimpleDateFormat(“yyyy-MM-dd HH:mm:ss”).format(new Date());

RateLimiter limiter = RateLimiter.create(1.0); // 这里的1表示每秒允许处理的量为1个

for (int i = 1; i <= 10; i++) {

limiter.acquire();// 请求RateLimiter, 超过permits会被阻塞

System.out.println(“call execute…” + i);

}

String end = new SimpleDateFormat(“yyyy-MM-dd HH:mm:ss”).format(new Date());

System.out.println(“start time:” + start);

System.out.println(“end time:” + end);

}

可以看到，我假定了每秒处理请求的速率为1个，现在我有10个任务要处理，那么RateLimiter就很好的实现了控制速率，总共10个任务，需要9次获取许可，所以最后10个任务的消耗时间为9s左右。那么在实际的项目中是如何使用的呢？？

实际项目中使用

@Service

public class GuavaRateLimiterService {

/每秒控制5个许可/

RateLimiter rateLimiter = RateLimiter.create(5.0);

/**

• 获取令牌

• @return

*/

public boolean tryAcquire() {

return rateLimiter.tryAcquire();

}

}

@Autowired

private GuavaRateLimiterService rateLimiterService;

@ResponseBody

@RequestMapping(“/ratelimiter”)

public Result testRateLimiter(){

if(rateLimiterService.tryAcquire()){

return ResultUtil.success1(1001,“成功获取许可”);

}

return ResultUtil.success1(1002,“未获取到许可”);

}

jmeter起10个线程并发访问接口，测试结果如下：

可以发现，10个并发访问总是只有6个能获取到许可，结论就是能获取到RateLimiter.create(n)中n+1个许可，总体来看Guava的RateLimiter是比较优雅的。本文就是简单的提了下RateLimiter的使用。

翻阅发现使用上述方式使用RateLimiter的方式不够优雅，尽管我们可以把RateLimiter的逻辑包在service里面，controller直接调用即可，但是如果我们换成：自定义注解+切面 的方式实现的话，会优雅的多，详细见下面代码：

自定义注解类

import java.lang.annotation.*;

/**

• 自定义注解可以不包含属性，成为一个标识注解

*/

@Inherited

@Documented

@Target({ElementType.METHOD, ElementType.FIELD, ElementType.TYPE})

@Retention(RetentionPolicy.RUNTIME)

public @interface RateLimitAspect {

}

自定义切面类

import com.google.common.util.concurrent.RateLimiter;

import com.simons.cn.springbootdemo.util.ResultUtil;

import net.sf.json.JSONObject;

import org.aspectj.lang.ProceedingJoinPoint;

import org.aspectj.lang.annotation.Around;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Pointcut;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.context.annotation.Scope;

import org.springframework.stereotype.Component;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletResponse;

import java.io.IOException;

@Component

@Scope

@Aspect

public class RateLimitAop {

@Autowired

private HttpServletResponse response;

private RateLimiter rateLimiter = RateLimiter.create(5.0); //比如说，我这里设置"并发数"为5

@Pointcut(“@annotation(com.simons.cn.springbootdemo.aspect.RateLimitAspect)”)

public void serviceLimit() {

}

@Around(“serviceLimit()”)

public Object around(ProceedingJoinPoint joinPoint) {

Boolean flag = rateLimiter.tryAcquire();

Object obj = null;

try {

if (flag) {

obj = joinPoint.proceed();

}else{

String result = JSONObject.fromObject(ResultUtil.success1(100, “failure”)).toString();

output(response, result);

}

} catch (Throwable e) {

e.printStackTrace();

}

System.out.println(“flag=” + flag + “,obj=” + obj);

return obj;

}

public void output(HttpServletResponse response, String msg) throws IOException {

response.setContentType(“application/json;charset=UTF-8”);

ServletOutputStream outputStream = null;

try {

outputStream = response.getOutputStream();

outputStream.write(msg.getBytes(“UTF-8”));

} catch (IOException e) {

e.printStackTrace();

} finally {

outputStream.flush();

outputStream.close();

}

}

总结

无论是哪家公司，都很重视高并发高可用的技术，重视基础，重视JVM。面试是一个双向选择的过程，不要抱着畏惧的心态去面试，不利于自己的发挥。同时看中的应该不止薪资，还要看你是不是真的喜欢这家公司，是不是能真的得到锻炼。其实我写了这么多，只是我自己的总结，并不一定适用于所有人，相信经过一些面试，大家都会有这些感触。

最后我整理了一些面试真题资料，技术知识点剖析教程，还有和广大同仁一起交流学习共同进步，还有一些职业经验的分享。

}

}

总结

无论是哪家公司，都很重视高并发高可用的技术，重视基础，重视JVM。面试是一个双向选择的过程，不要抱着畏惧的心态去面试，不利于自己的发挥。同时看中的应该不止薪资，还要看你是不是真的喜欢这家公司，是不是能真的得到锻炼。其实我写了这么多，只是我自己的总结，并不一定适用于所有人，相信经过一些面试，大家都会有这些感触。

最后我整理了一些面试真题资料，技术知识点剖析教程，还有和广大同仁一起交流学习共同进步，还有一些职业经验的分享。

[外链图片转存中…(img-Pf1MiR5c-1716575098044)]