Skip to main content

Redis获取六位不重复数字(邀请码)

· 3 min read
Kuizuo

需求

针对每一个用户(用户量在 10w 以下)随机生成的邀请码(仅限六位数字),且不重复

思考

如果能把这个不重复条件去除,那么只需要使用Math.random然后取小数点后六位就行了,但可惜要求就是不能重复, 要是重复还得了,到时候注册的时候都不知道奖励给那个邀请码账号。同时还要求邀请码在六位且数字,这就导致即使随机生成的,会有一定的可能出现相同的邀请码。

解决方案

方案 1

先随机生成一个六位随机数字,然后在存的时候判断数据库是否存在该邀请码,如果存在那么就重新生成一个,直到该邀请码不存在,便存入。

优点:方便,如果用户量不大,完全可以 缺点:用户量上来的情况下,判断邀请码是否存在有可能需要一段时间,并且由于需要判断,故性能欠缺

方案 2

利用 redis 的 set 数据类型,先将所有的邀请码存入到 set 中,然后通过 srandmember 随机获取一个数值,在通过 srem 删除该元素即可。

或者也可以通过 list 队列,将预先随机生成的六位不重复数字的所有集合统统添加到队列中,然后获取的时候通过 rpop 或 lpop 获取

优点:相当于空间换时间,无需判断,后期即便用户量上来的,也完全可以重新生成一批(七位或字母)重新导入

缺点:过于依赖 Redis,redis 服务一旦停止,便无法正常获取数据。

实现

既然想都想了,那怎么能不实现呢。我这边仅仅是一个测试 Demo,利用的是方案 2,通过 set 数据类型进行获取相关代码如下

预先存入数据

let key = 'code';
function genCode() {
let num = 999999;
for (let i = 100000; i < num; i++) {
client.sadd(key, i, function (err, data) {});
}
console.log('数据导入完毕');
}
genCode();

获取数据并删除

// 输出所有成员
client.smembers(key, function (err, data) {
console.log(data);
});

// 随机获取一个数据
client.srandmember(key, function (err, data) {
console.log(data);
client.srem(key, data, function (err, data) {});
});

整体耗时不会超过 3 分钟

通过console.time()获取数据耗时如下

default: 0.174ms
Loading Comments...