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

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

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值