分享关于限流器的10大Java面试问题。
微信搜索关注《Java学研大本营》
速率限制是指对用户在一定时间段内向系统发出的请求次数进行控制，以保证系统的稳定性和资源的公平分配。
可以使用令牌桶或滑动日志方法。
public class RateLimiter {
private final long maxRequests;
private long lastRequestTime = System.currentTimeMillis();
private long currentRequests = 0;
public RateLimiter(long maxRequestsPerSecond) {
this.maxRequests = maxRequestsPerSecond;
}
public synchronized boolean allowRequest() {
long currentTime = System.currentTimeMillis();
if (currentTime - lastRequestTime > 1000) {
lastRequestTime = currentTime;
currentRequests = 0;
}
if (currentRequests < maxRequests) {
currentRequests++;
return true;
}
return false;
}
}
系统以一定的频率向桶里放入令牌。每个请求都要消耗一个令牌。如果桶里没有令牌，请求就会被拒绝。
Redis，凭借其原子操作和过期键，可以在分布式系统中高效地跟踪记录请求次数或令牌数量。
Redis 可用于限制速率
使用一个集中式存储，如Redis，或一个分布式配置系统，如ZooKeeper，来协调多个实例之间的速率限制。
Redis 可处理分布式速率限制
ZooKeeper 可处理分布式速率限制
有状态的限流器会保存状态（比如请求次数），而无状态的限流器只根据当前的数据做出决策，不存储过去的信息。
无状态的限流器不会在请求之间保留任何状态，也就是说它不会记住之前的请求。它只是根据当前请求的信息做出决策。无状态限流器的一种常用方法是使用JWT（JSON Web Token）或类似的令牌，令牌里面包含了必要的信息。
下面是使用JWT的简单示例：
客户端请求访问并接收一个JWT，其中包含过期时间和允许的最大请求数。
对于每个请求，客户端发送JWT。
服务器验证JWT，并检查过期时间和已发出的请求数。
如果客户端在时间范围内超过了请求数，服务器拒绝请求。
下面是基本的实现：
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import io.jsonwebtoken.Claims;
public class StatelessRateLimiter {
private static final String SECRET_KEY = "mySecretKey";
private static final int MAX_REQUESTS = 10;
private static final int ONE_HOUR = 3600000;
public String generateToken() {
long expirationTime = System.currentTimeMillis() + ONE_HOUR;
return Jwts.builder()
.setSubject("rateLimitToken")
.claim("requests", 0)
.setExpiration(new Date(expirationTime))
.signWith(SignatureAlgorithm.HS512, SECRET_KEY)
.compact();
}
public boolean allowRequest(String token) {
try {
Claims claims = Jwts.parser()
.setSigningKey(SECRET_KEY)
.parseClaimsJws(token)
.getBody();
int requests = claims.get("requests", Integer.class);
if (requests < MAX_REQUESTS) {
claims.put("requests", requests + 1);
return true;
}
return false;
} catch (Exception e) {
return false;
}
}
}
这是一个简化的例子。在实际场景中，你需要处理令牌的更新，确保令牌的安全存储，并管理其他安全方面。这里使用的JWT库是
jjwt
。
用一个列表或双端队列来存储请求的时间戳。保证在每个时间窗口内的请求次数不超过限制。
import java.util.Deque;
import java.util.LinkedList;
public class SlidingWindowRateLimiter {
private final Deque<Long> timestamps;
private final int maxRequests;
private final long windowSizeInMillis;
public SlidingWindowRateLimiter(int maxRequests, long windowSizeInMillis) {
this.timestamps = new LinkedList<>();
this.maxRequests = maxRequests;
this.windowSizeInMillis = windowSizeInMillis;
}
public synchronized boolean allowRequest() {
long currentTime = System.currentTimeMillis();
// 移除当前窗口外的时间戳
while (!timestamps.isEmpty() && timestamps.peekFirst() < currentTime - windowSizeInMillis) {
timestamps.pollFirst();
}
// 检查是否添加一个新请求会超过最大限制
if (timestamps.size() < maxRequests) {
timestamps.addLast(currentTime);
return true;
}
return false;
}
public static void main(String[] args) {
SlidingWindowRateLimiter limiter = new SlidingWindowRateLimiter(5, 1000); // 每秒5个请求
for (int i = 0; i < 10; i++) {
System.out.println(limiter.allowRequest()); // 前5个为true，后5个为false
try {
Thread.sleep(200); //睡眠200毫秒
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
}
在这个例子中，
SlidingWindowRateLimiter
限制了在一定时间窗口（
windowSizeInMillis
）内的最大请求次数（
maxRequests
）。
Deque
用于存储请求的时间戳。每次判断是否可以接受新请求时，它会先删除已经超出当前窗口的时间戳，然后再把
Deque
的长度和最大请求次数对比。
在API网关层实现限流器或使用集中式存储来实现分布式速率限制方法。
要求尽可能降低延迟，应对海量的请求，并在监控和执行限制的过程中不影响系统性能。
使用HTTP头，如
X-RateLimit-Limit
，
X-RateLimit-Remaining
，和
X-RateLimit-Reset
来传达速率限制的细节。
速率限制是后端工程广阔领域中的一项重要技术，它能够保证系统的稳定性和资源的公平分配。我们已经了解了Java提供的各种工具和技术，可以用来实现有效的速率限制策略。无论你是准备面试，还是想要优化你的后端系统，掌握速率限制的细节都是非常必要的。建议你深入学习并进行实验。
《Java从入门到精通（第7版）》从初学者角度出发，通过通俗易懂的语言、丰富多彩的实例，详细讲解了使用Java语言进行程序开发需要掌握的知识。全书分为4篇共24章，内容包括初识Java，开发工具（IDEA、Eclipse），Java语言基础，流程控制，数组，类和对象，继承、多态、抽象类与接口，包和内部类，异常处理，字符串，常用类库，集合类，枚举类型与泛型，lambda表达式与流处理，I/O（输入/输出），反射与注解，数据库操作，Swing程序设计，Java绘图，多线程，并发，网络通信，飞机大战游戏，MR人脸识别打卡系统。书中所有知识都结合具体实例进行讲解，涉及的程序代码都给出了详细的注释，这可以帮助读者轻松领会Java程序开发的精髓，并快速提高开发技能。
《Java从入门到精通（第7版）（软件开发视频大讲堂）》
https://item.jd.com/14067396.html
精彩回顾
一文搞懂Java面向对象编程中的封装机制
一文搞懂Java面向对象编程中的继承机制
详解Java中的装饰器
5个Java开发者不可不知的编程库
用好这10个Spring Cloud功能，使用Java轻松开发微服务
微信搜索关注《Java学研大本营》
访问【IT今日热榜】，发现每日技术热