限制每个用户访问接口的次数(Java)

 

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

private boolean checkSendMobile2(String mobile){

        int limitCount = 0;

        Object obj = UserCacheManager.getValue("SEND_SMS_TYPE",

                mobile);

        String limitCountKey = mobile+"_limitCount";//每个手机号码就代表是一个用户

        if(UserCacheManager.getValue("SEND_SMS_LIMIT_COUNT", limitCountKey)==null){

            limitCount = 1;// 记录第一次

        }else{

            limitCount = (int) UserCacheManager.getValue("SEND_SMS_LIMIT_COUNT", limitCountKey);

        }

         

        if (obj instanceof Long) {

            long time = (Long) obj;

            if(System.currentTimeMillis() - time < 1 60 1000){

                //如果是在一分钟之内 则判断访问次数

                limitCount++;

                UserCacheManager.put("SEND_SMS_LIMIT_COUNT", limitCountKey, limitCount);

                if(limitCount>3){

                    //如果大于三次 则调用次数过多

                    return true;

                }else{

                    return false;

                }

            }else{

                //超过一分钟

                limitCount = 1;//次数重新赋值1

                //同时保存用户访问次数和时间

                UserCacheManager.put("SEND_SMS_TYPE", mobile, new Date().getTime());

                UserCacheManager.put("SEND_SMS_LIMIT_COUNT", limitCountKey, limitCount);

                return false;

            }

        else {

            //没有保存用户访问时间, 则说明第一次访问

            UserCacheManager.put("SEND_SMS_TYPE", mobile, new Date().getTime());

            UserCacheManager.put("SEND_SMS_LIMIT_COUNT", limitCountKey, limitCount);

            return false;

        }

    }

map<用户,List<时间(long)>>
比如你吧用户信息也传到这个方法里面去,map.get就能找到你最近访问的时间记录了,几点几分第一次,几点几分第二次,
还是按照时间先后顺序排列,都已经这样了。
for(List<时间(long)>){
if(list.get(i)-当前时间<三分钟){
    ++n;
}
}
if(n>3){
throw ;
}

 

好的,下面是一个简单的使用过滤器限制每个用户访问接口次数Java代码示例: 首先,创建一个名为 `ApiAccessLimitFilter` 的过滤器类,代码如下: ```java import javax.servlet.*; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.IOException; public class ApiAccessLimitFilter implements Filter { private static final int MAX_ACCESS_COUNT = 10; // 每个用户最大访问次数 private static final String CACHE_PREFIX = "api_access_count_"; // 缓存键前缀 private CacheManager cacheManager; // 缓存管理器 @Override public void init(FilterConfig filterConfig) throws ServletException { // 初始化缓存管理器,这里使用Ehcache作为缓存实现 cacheManager = CacheManager.create(); cacheManager.addCache("api_access_count_cache"); } @Override public void destroy() { // 关闭缓存管理器 cacheManager.shutdown(); } @Override public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException { HttpServletRequest httpRequest = (HttpServletRequest) request; HttpServletResponse httpResponse = (HttpServletResponse) response; String userId = getUserId(httpRequest); // 获取用户ID String cacheKey = CACHE_PREFIX + userId; // 构造缓存键 // 从缓存中获取用户访问计数器 Element element = cacheManager.getCache("api_access_count_cache").get(cacheKey); int accessCount = element != null ? (int) element.getValue() : 0; if (accessCount >= MAX_ACCESS_COUNT) { // 如果用户访问次数超过限制 httpResponse.setStatus(429); // 返回 HTTP 429 Too Many Requests 状态码 return; } // 用户访问次数加1,并更新缓存 accessCount++; Element newElement = new Element(cacheKey, accessCount); cacheManager.getCache("api_access_count_cache").put(newElement); chain.doFilter(request, response); // 继续处理请求 } private String getUserId(HttpServletRequest request) { // 这里假设从请求头中获取用户ID return request.getHeader("user-id"); } } ``` 以上代码中,我们使用了一个 `CacheManager` 对象来管理缓存,这里使用了开源的 Ehcache 实现。在 `init` 方法中初始化了 `CacheManager` 对象和一个名为 `api_access_count_cache` 的缓存实例。在 `doFilter` 方法中,我们首先通过 `getUserId` 方法获取用户ID,然后构造一个缓存键 `cacheKey`,并从缓存中获取该用户访问计数器。如果用户访问次数超过了限制,直接返回 HTTP 429 Too Many Requests 状态码。否则,将用户访问次数加1,并更新缓存。最后,调用 `chain.doFilter(request, response)` 方法继续处理请求。 接下来,在 `web.xml` 文件中注册该过滤器,代码如下: ```xml <filter> <filter-name>ApiAccessLimitFilter</filter-name> <filter-class>com.example.ApiAccessLimitFilter</filter-class> </filter> <filter-mapping> <filter-name>ApiAccessLimitFilter</filter-name> <url-pattern>/api/*</url-pattern> </filter-mapping> ``` 以上代码中,我们将该过滤器注册到了所有以 `/api/` 开头的 URL 上,即只有访问 `/api/` 开头的接口时才会触发该过滤器。 这样,我们就成功实现了使用过滤器限制每个用户访问接口次数的功能。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值