需求描述:
产品狗说要做一个注册码生成模块,生成一批注册码,提供给受邀的用户注册网站。
需求分析
注册码功能并不少见(想去xxx论坛的孩子们天天吵着要注册码……),首先我们来简单分析一下产品狗的这个需求。
产品狗说要做一个注册码生成模块,生成一批注册码,提供给受邀的用户注册网站
产品狗随口一句话,里面包含这么多信息:
- 自动产生注册码
- 每次产生一个注册码或者是多个注册码(数量前台应该可以输入)
- 注册码不能重复(这个是必须的)
- 注册码得有一个标识,已使用或者未使用。
注册码表的设计看上去是这个样子的:
var _getInviteCode = function () {
var inviteCodeSchema = new Schema({
code: {type: String, required: true, unique: true},
used: {type: Boolean, default: false}
}, {
timestamps: {
createdAt: 'created_at',
updatedAt: 'updated_at'
}
});
return mongoose.model('inviteCode', inviteCodeSchema);
};
其实就俩主要字段code
和used
。
需求解决
分析一下注册码生成机制
我总结了一下主要有以下几种方案:
-
自己定义一个字符池,将允许的符号放到池子里,随机取n个字符组成注册码。
缺点:自己写的算法,为了防止不重复,每次需要跟数据库里已有的值进行对比,效率慢。 -
使用GUID
缺点:字符串较长,占用空间较大,不美观。 -
主键或主键+随机码的形成
缺点:需要先产生主键,而且与数据库类型关系大,扩展性和复用性较差。 -
使用系统日期
缺点:高并发的时候,可能重复
目前用的比较多的是第三种方式,衍生出来可以是:将唯一ID映射成base26\base36。
我这里推荐一个项目:https://github.com/dylang/shortid 可以有效快速的生成不重复的注册码(可用于游戏开发)。
解决循环插入数据库的问题
注册码生成方案有了,但是在具体编码的过程中又遇到了新的问题:如何批量保存一批注册码呢
Mongoose本身支持批量插入数据,所以这个问题很好解决:
router.get('/generate', function (req, res, next) {
var number = Number(req.query.num);
var codeArr = new Array();
for(var i = 0; i< number; i++) {
codeArr.push({
code: shortid.generate()
});
}
InviteCode.collection.insert(codeArr, function (err, docs) {
if(err){
next(err);
}else{
res.redirect('/admin/inviteCode');
}
});
});