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

Java code
?

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、付费专栏及课程。

余额充值