API调用控制器,对接口的访问进行控制

Java限制一个方法一段时间内的执行次数

开发中的场景,微信链接分享,对于新用户的访问,在一定的时间,一天或者一周只统计一次用户的访问量。前端传入微信号的openid进行比较。


import org.springframework.boot.test.context.SpringBootTest;

import java.util.HashMap;
import java.util.Map;


@SpringBootTest
public class APICallLimiter {
    private static final int MAX_CALLS_PER_WEEK = 1; // 每周最大调用次数
    private static final long WEEK_IN_MILLIS = 7 * 24 * 60 * 60 * 1000; // 一周的毫秒数
    private Map<String, Integer> callCounterMap = new HashMap<>(); // 用户请求次数计数器
    private Map<String, Long> lastCallTimeMap = new HashMap<>(); // 用户上次请求时间戳

    public boolean canMakeAPICall(String userId) {
        long currentTime = System.currentTimeMillis();
        int callCount = callCounterMap.getOrDefault(userId, 0);
        long lastCallTime = lastCallTimeMap.getOrDefault(userId, 0L);

        if (currentTime - lastCallTime < WEEK_IN_MILLIS) { // 如果时间间隔过短,拒绝处理请求
            return false;
        }

        if (callCount >= MAX_CALLS_PER_WEEK) { // 如果请求次数超过限制,拒绝处理请求
            return false;
        }

        callCounterMap.put(userId, callCount + 1); // 更新请求次数
        lastCallTimeMap.put(userId, currentTime); // 更新上次请求时间戳

        return true;
    }

    public static void main(String[] args) {
        APICallLimiter limiter = new APICallLimiter();
        String userId = "user1";

        for (int i = 0; i < 15; i++) {
            if (limiter.canMakeAPICall(userId)) {
                System.out.println("允许处理请求");
            } else {
                System.out.println("拒绝处理请求");
            }
            try {
                Thread.sleep(1000); // 暂停1秒
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值