注册锁

src/modules/sign/sign.controller.js

@@ -5,7 +5,9 @@ const User = require('../user/user.model');
 const { createNewUser } = require('../user/user.handler');
 const Code = require('../msgCode/msgCode.controller');
 const Fund = require('../fund/fund.model');
+const Lock = require('../../utils/mutexlock');
 
+const lock = new Lock();
 const CODE_TYPE = Code.CODE_TYPE;
 const msgCodeSignIn = async function (phone, code) {
     const validCode = await Code.verifyCode({ id: phone, code, type: CODE_TYPE['msgSignIn'] });
@@ -13,6 +15,7 @@ const msgCodeSignIn = async function (phone, code) {
         phone,
     }).exec() : null;
 };
+const phoneRegCN = /^\+86-[1]([3-9])[0-9]{9}$/;
 
 module.exports = {
     checkExist: async function (req, res) {
@@ -46,7 +49,7 @@ module.exports = {
                 const token = createToken({
                     userId: user.id
                 }, JWT_SECRET, {});
-                res.json({
+                return res.json({
                     code: 0,
                     msg: 'success',
                     data: {
@@ -58,7 +61,7 @@ module.exports = {
                     }
                 });
             } else {
-                res.json({
+                return res.json({
                     code: 1,
                     error: 'unknow user',
                     msg: '账号或密码错误'
@@ -69,42 +72,49 @@ module.exports = {
     signUpReqInviter: inject(function(SysConfig) {
         const { JWT_SECRET, PWD_HASH_SALT } = SysConfig;
         return async (req, res) => {
-            return res.json({
-                code: 404,
-                msg: '暂停注册'
-            });
+            // return res.json({
+            //     code: 404,
+            //     msg: '暂停注册'
+            // });
             let { phone, code, inviter, password } = req.body;
             if (!phone || !code || !password || !inviter) return res.json({
                 code: 1,
                 msg: 'Missing required parameters'
             });
-            const exist = await User.exists({ phone });
-            if (exist) return res.json({
-                code: 1,
-                msg: ''
+            phone = String.prototype.trim.call(phone);
+            const isValidPhone = phone.startsWith('+86-') ? phoneRegCN.test(phone) : true; //国际手机暂不检测
+            if (!isValidPhone) return res.json({
+                code: 5,
+                msg: 'Invalid phone',
+            });
+            const isValidCode = await Code.verifyCode({ id: phone, code, type: CODE_TYPE['signUp'] });
+            if (!isValidCode) {
+                return res.json({
+                    code: 2,
+                    msg: 'Invalid code',
                 });
-            let validInviter = '';
-            if (inviter === 'AlUx9h!w0)y9') {
-                validInviter = { id: 'AlUx9h!w0)y9' };
             }
-            else {
+            let validInviter = '';
             const invitePhone = '+86-' + parseInt(inviter, 36);
             validInviter = await User.findOne({ phone: invitePhone }).catch(null);
             if (!validInviter) return res.json({
                 code: 4,
                 msg: '邀请人不存在'
             });
-            }
-            const isCodeValid = await Code.verifyCode({ id: phone, code, type: CODE_TYPE['signUp'] });
+            const exist = await User.exists({ phone });
+            if (exist || (!exist && !lock.acquire(phone))) return res.json({
+                code: 1,
+                msg: '重复注册'
+            });
             password = Crypto.hash(password, { salt: PWD_HASH_SALT });
             const newInviteCode = typeof phone === 'string' && parseInt(phone.split('-')[1]).toString(36);
-            if (isCodeValid) {
             await createNewUser({
                 phone,
                 inviter: validInviter.id,
                 password,
                 inviteCode: newInviteCode
             }).then(user => {
+                lock.unlock(phone);
                 User.findByIdAndUpdate(validInviter.id, { $push: { group: user.id } }).catch(err => {
                     mylog.error('将被邀请人加入到邀请人队列失败 uid', user.id, err && err.errmsg);
                 });
@@ -112,73 +122,21 @@ module.exports = {
                 const token = createToken({
                     userId: user.id
                 }, JWT_SECRET, {});
-                    res.json({
+                return res.json({
                     code: 0,
                     msg: 'success',
-                        data: {
-                            //刚注册，需要返回的应该只有token，其他的没意义
-                            token
-                        }
+                    data: { token }
                 });
-                })
-                .catch(err => {
-                    mylog.error(err);
-                    res.json({
+            }).catch(err => {
+                mylog.error('[SignUpWI]创建用户失败', err);
+                return res.json({
                     code: 3,
                     msg: JSON.stringify(err),
                 });
             });
-            } else {
-                res.json({
-                    code: 2,
-                    msg: 'Invalid code',
-                });
-            }
-        };
-    }),
-    signUp: inject(function(SysConfig) {
-        const { JWT_SECRET, PWD_HASH_SALT } = SysConfig;
-        return async (req, res) => {
-            let { phone, code, inviter, password } = req.body;
-            if (!phone || !code || !password) return res.json({
-                code: 1,
-                msg: 'Missing required parameters'
-            });
-            const isCodeValid = await Code.verifyCode({ id: phone, code, type: CODE_TYPE['signUp'] });
-            password = Crypto.hash(password, { salt: PWD_HASH_SALT });
-            if (isCodeValid) {
-                createNewUser({
-                    phone,
-                    inviter,
-                    password
-                }).then(user => {
-                    const token = createToken({
-                        userId: user.id
-                    }, JWT_SECRET, {});
-                    res.json({
-                        code: 0,
-                        msg: 'success',
-                        data: {
-                            //刚注册，需要返回的应该只有token，其他的没意义
-                            token
-                        }
-                    });
-                })
-                .catch(err => {
-                    mylog.error(err);
-                    res.json({
-                        code: 3,
-                        msg: JSON.stringify(err),
-                    });
-                });
-            } else {
-                res.json({
-                    code: 2,
-                    msg: 'Invalid code',
-                });
-            }
         };
     }),
+    signUp: this.signUpReqInviter, // 暂时不用，先用要求邀请人的代替
     forgetPasswd: inject(function(SysConfig) {
         const { PWD_HASH_SALT } = SysConfig;
         return async (req, res) => {

src/utils/mutexlock.js

@@ -0,0 +1,21 @@
+class MLock {
+    constructor (props) {
+        Object.defineProperty(this, 'map', {
+            value: new Map(),
+            writable: false
+        });
+    }
+    acquire (key) {
+        if (this.map.has(key)) {
+            return false;
+        }
+        this.map.set(key, true);
+        return true;
+    }
+    unlock (key) {
+        this.map.delete(key);
+        return true;
+    }
+}
+
+module.exports = MLock;