springboot项目限制接口外部访问频率每秒钟接口访问次数不得超过1次(使用ConcurrentHashMap实现)

矫情吗；* 2023-10-10 16:49 44阅读 0赞

使用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次的接口访问频率限制。需要注意的是，如果有多个请求同时访问同一个接口，可能会出现线程安全问题，需要进行处理。