使用Redis做一个简单的新好友数量限制规则

傻了,更新键值对的时候没有把过期时间重新设置回去,导致过期时间都被刷新了。具体修改(20行):
redisClient.setAsync(redisKey, data);
改为:
redisClient.setAsync(redisKey, data, 'EX', secondGap);
//--------------------------------20170519更新分割线--------------------------------
线上项目碰到有人恶意发送广告消息导流,并且 APP 本身是不需要好友申请就能发送消息的,因此需要做一个简单的创建新会话限制。
功能比较简单明确,但是如果使用普通的数据库,感觉有点小题大做。如果单纯自己构造一个内存数组又容易丢失数据,所以想到用 Redis 来做一个计数器。

  1. 幼年期,最简单的版本是直接使用 Redis 的基础命令,用 userId 做 key,直接叠加:

    INCR userId

    但是这种只能计算一个累加和,无法做到按时间或者按日期这种细粒度规则限制。

  2. 成长期,利用 expire,做一个定时的策略。给一个键值对做一个超时时间(例如一小时):

    SET userId 1 EX 3600

    每次申请创建会话则加一,超过固定上限就禁止叠加,并且一小时后计时器默认归零,不需要手动再去创建定时器来做清零操作。

  3. 成熟期,利用 expire,根据规则做一个数字编码,减少 Redis 键值创建和删除的次数,同样是一个小时不能超过十个新的会话。

    // 是否能创建新的会话
    function canCreateConversation(userId) {
        var redisKey = userId; // 用userId做 redis 的 key
        var convsPerHour = 10; // 新会话的个数,一小时限制在十个
        var now = new Date();
        var hour = now.getHours();
        var miniData = hour * 100 + 1; // 当前小时的最小值,每个小时数字不会超过10,所以将小时数记在第三位
        var maxData = hour * 100 + convsPerHour; // 当前小时的最大值,同上
        // 计算键的过期时间
       var tomorrow = new Date((new Date()).setDate(now.getDate() + 1)).setHours(0, 0, 0, 0); // 明天凌晨
       var secondGap = parseInt((tomorrow.getTime() - now.getTime()) / 1000); // 每天十二点刷新数据,方便之后做每天创建会话数量的限制,这里存在一个时区的问题
        return redisClient.getAsync(redisKey).then(function(data) {
            if (data) {
                data = parseInt(data);
                if (data >= maxData) {
                    // 这个小时已经超过最大创建限制了
                    return Promise.resolve(false);
                } else {
                    data = data < miniData ? miniData : data + 1;
                    redisClient.setAsync(redisKey, data, 'EX', secondGap);
                    return Promise.resolve(true);
                }
            } else {
                redisClient.setAsync(redisKey, miniData, 'EX', secondGap);
                return Promise.resolve(true);
            }
        });
    }
    
  4. 完全期,在上面的规则上,加入一个每天创建新会话的上限:

    也就是数字编码的时候,在表示小时数的位之上,再加入一段来记载每天创建的会话数。

你可能感兴趣的