From 4da8a11bead2c92c830af001f11e2a62e823d2cd Mon Sep 17 00:00:00 2001
From: chaosBreaking <x491807573@qq.com>
Date: Wed, 9 Oct 2019 22:47:33 +0800
Subject: [PATCH] msg code

---
 src/modules/msgCode/msgCode.controller.js | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/src/modules/msgCode/msgCode.controller.js b/src/modules/msgCode/msgCode.controller.js
index f1c21c4..553e4df 100644
--- a/src/modules/msgCode/msgCode.controller.js
+++ b/src/modules/msgCode/msgCode.controller.js
@@ -26,10 +26,10 @@ module.exports = {
             id: phone,
             type: CODE_TYPE[type]
         });
-        if (exist) return {
-            code: 1,
-            error: '请勿多次请求'
-        };
+        // if (exist) return {
+        //     code: 1,
+        //     error: '请勿多次请求'
+        // };
         if (phone && type && validCodeType) {
             const code = Crypto.randomNumber({length: 6});
             const [ sms, save ] = await Promise.all([
@@ -42,7 +42,7 @@ module.exports = {
                     id: phone,
                     code,
                     type: CODE_TYPE[type]
-                })
+                }, exist)
             ]).catch(err => {
                 mylog.error(err);
                 return [null, null];
@@ -57,8 +57,13 @@ module.exports = {
             error: 'Invalid param'
         };
     },
-    async saveCode (data) {
-        const res = await (new Code({ ...data })).save();
+    async saveCode (data, exist) {
+        let res;
+        if (!exist) {
+            res = await (new Code({ ...data })).save();
+        } else {
+            res = await Code.findOneAndUpdate({ id: data.id, type: data.type }, { code: data.code });
+        }
         if(!res) {
             return false;
         } else {