Redis限制N小时内每用户Id最多只能登录M次解决方案

package com.zxl.redis.rediscode;


import redis.clients.jedis.Jedis;

import java.util.List;

/**
 * @author xiaoliang
 * @date 2020/5/12 21:30
 * @since V1.0
 */
public class LoginValidate {
    //设置成10s是因为方便测试
    //周期时间为毫秒
    final static int EXPIRE_TIME_MILLISECOND = 10*1000;
    //一个周期内最多的操作次数
    final static int MOST_TIMES_IN_TIME_LIMIT = 5;

    public static void main(String[] args) {
        //创建Jedis客户端
        Jedis jedis = new Jedis("192.168.0.3",6379);
        //操作数据
        jedis.select(0);
        //模拟登入用户
         String userId="zxl";
        loginValidateWithQueue(jedis,userId);
        loginValidateWithQueue(jedis,userId);
        loginValidateWithQueue(jedis,userId);
        try {
            Thread.sleep(5000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        loginValidateWithQueue(jedis,userId);
        loginValidateWithQueue(jedis,userId);
        loginValidateWithQueue(jedis,userId);
        try {
            Thread.sleep(6000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        loginValidateWithQueue(jedis,userId);
        loginValidateWithQueue(jedis,userId);
        //取出用户最后五次的操作时间
        List<String> values=jedis.lrange(userId,0,-1);
        for (int i = 0; i < values.size(); i++) {
            System.out.printf(values.get(i) + "\n");
        }
        //删除用户列表
        jedis.del(userId);
        jedis.close();
    }

    /**
     * 用Redis和任意语言实现一段恶意登录保护的代码,限制1小时内每用户Id最多只能登录5次
     *
     * @param jedis
     * @param userId
     * @return
     */
    public static boolean loginValidateWithQueue(Jedis jedis, String userId) {
        Long currentTime = System.currentTimeMillis();
        //判断List列表长度是否超过操作次数
        if (jedis.llen(userId) < MOST_TIMES_IN_TIME_LIMIT) {
            //将当前操作时间放入到用户list中
            jedis.rpush(userId, currentTime.toString());
            System.out.println(currentTime + " 登入成功!" + jedis.llen(userId));
            return true;
        } else {
            //如果超过操作次数则取出list中最早操作时间,即list下标为0的
            String earliestTimeStr = jedis.lindex(userId, 0);
            //将字符串转为长整形
            Long earliestTime = Long.valueOf(earliestTimeStr);
            //判断最早操作时间是否在周期内,如果在周期内说明在这个周期内操作数已到阀值不允许再操作。
            if (currentTime - earliestTime <= EXPIRE_TIME_MILLISECOND) {
                System.out.println(EXPIRE_TIME_MILLISECOND/1000 + " 秒内只能操作" + MOST_TIMES_IN_TIME_LIMIT + " 次");
                return false;
            } else {
                //如果最早操作时间不再周期,说明可以继续操作。
                //为了使用户列表长度为操作次数。需要删除最早的操作时间,保存本次操作时间。
                //删除列表尾段数据,即最早操作时间。
                jedis.lpop(userId);
                //插入当前操作时间
                jedis.rpush(userId, currentTime.toString());
                System.out.println(currentTime + " 登入成功!" + jedis.llen(userId));
                return true;
            }
        }
    }
}

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
可以实现。首先,我们需要在 Redis 中维护一个 key-value 键值对,每个用户对应一个记录其发送验证码次数的计数器。每用户发送验证码时,我们会从 Redis 中获取该用户的计数器,判断其是否达到了十限制。如果没有达到限制,则允许用户发送验证码,并递增计数器。如果超过限制,则拒绝用户发送验证码。 使用 Redis 进行实现,你需要用到以下命令: 1. `INCR key`:让键 key 的值加一,如果 key 不存在,则创建并把值初始化为 0,返回增加后的结果。 2. `EXISTS key`:返回键 key 是否存在于 Redis 中。 3. `TTL key`:获取 key 的剩余生存时间。 4. `SET key value [EX seconds] [PX milliseconds] [NX|XX]`:将键 key 的值设为 value,如果 key 已经存在,则用新的 value 进行覆盖。同时可以指定键的过期时间。 你可以通过以下代码实现: ```python import redis class RedisUtils: def __init__(self, host, port, db): self.redis_conn = redis.Redis(host=host, port=port, db=db) def send_code_available(self, user_id): key = f"send_code_times_{user_id}" count = self.redis_conn.get(key) if count and int(count) >= 10: return False self.redis_conn.incr(key) self.redis_conn.expire(key, 3600) return True redis_conn = RedisUtils("localhost", 6379, 0) # Redis 连接 user_id = "123456" # 用户 ID if redis_conn.send_code_available(user_id): # 发送验证码 pass else: print("一小时内不能发送超过十验证码") ```
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值