傻了,更新键值对的时候没有把过期时间重新设置回去,导致过期时间都被刷新了。具体修改(20行):
redisClient.setAsync(redisKey, data);
改为:
redisClient.setAsync(redisKey, data, 'EX', secondGap);
//--------------------------------20170519更新分割线--------------------------------
线上项目碰到有人恶意发送广告消息导流,并且 APP 本身是不需要好友申请就能发送消息的,因此需要做一个简单的创建新会话限制。
功能比较简单明确,但是如果使用普通的数据库,感觉有点小题大做。如果单纯自己构造一个内存数组又容易丢失数据,所以想到用 Redis 来做一个计数器。
-
幼年期,最简单的版本是直接使用 Redis 的基础命令,用 userId 做 key,直接叠加:
INCR userId
但是这种只能计算一个累加和,无法做到按时间或者按日期这种细粒度规则限制。
-
成长期,利用 expire,做一个定时的策略。给一个键值对做一个超时时间(例如一小时):
SET userId 1 EX 3600
每次申请创建会话则加一,超过固定上限就禁止叠加,并且一小时后计时器默认归零,不需要手动再去创建定时器来做清零操作。
-
成熟期,利用 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); } }); }
-
完全期,在上面的规则上,加入一个每天创建新会话的上限:
也就是数字编码的时候,在表示小时数的位之上,再加入一段来记载每天创建的会话数。