使用Java内置的ConcurrentHashMap来实现接口访问频率限制。具体实现方法如下:
创建一个ConcurrentHashMap对象,用于存储每个请求的访问次数和时间戳。我们可以将其定义为一个静态变量,确保在多个请求之间共享。
private static ConcurrentHashMap<String, Long> requestCountMap = new ConcurrentHashMap<>();
创建一个限制方法访问频率的方法,该方法需要在接口方法中调用。该方法需要获取当前请求的IP地址和URI,并计算该请求与上一次请求的时间间隔。如果时间间隔小于1秒钟,则返回false,否则返回true。如果返回true,则将当前请求的时间戳存储到ConcurrentHashMap中,并将访问次数设置为1。如果返回false,则将访问次数加1。
private boolean limitRequestFrequency(HttpServletRequest request) {
String ip = request.getRemoteAddr();
String uri = request.getRequestURI();
String key = ip + ":" + uri;
Long lastRequestTime = requestCountMap.get(key);
long now = System.currentTimeMillis();
if (lastRequestTime != null && (now - lastRequestTime) < 1000) {
return false;
}
requestCountMap.put(key, now);
return true;
}
在接口方法中调用限制访问频率的方法。如果该方法返回false,则说明请求未通过访问频率限制,需要返回429状态码和提示信息;否则,说明请求通过了访问频率限制,可以继续执行接口方法。
@GetMapping("/hello")
public String hello(HttpServletRequest request, HttpServletResponse response) {
if (!limitRequestFrequency(request)) {
response.setStatus(429);
return "Too many requests";
}
return "Hello, World!";
}
在以上实现中,我们使用Java内置的ConcurrentHashMap来实现了每秒钟接口访问次数不得超过1次的接口访问频率限制。需要注意的是,如果有多个请求同时访问同一个接口,可能会出现线程安全问题,需要进行处理。