From bf43bfbd92b64fc0f320467314d7e9e3cdffebd7 Mon Sep 17 00:00:00 2001
From: hanhh <18330117883@163.com>
Date: Wed, 8 Sep 2021 12:26:05 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=BC=80=E5=90=AF=E4=B8=A4?=
 =?UTF-8?q?=E6=AD=A5=E5=A4=B1=E8=B4=A5=E6=8F=90=E9=86=92?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/pages/setting/security.tsx | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/pages/setting/security.tsx b/src/pages/setting/security.tsx
index b2925258..e689bbcd 100644
--- a/src/pages/setting/security.tsx
+++ b/src/pages/setting/security.tsx
@@ -55,6 +55,7 @@ const SecuritySettings = ({ user, userChange }: any) => {
   };
 
   const completeTowFactor = () => {
+    setLoading(true);
     request
       .put(`${config.apiPrefix}user/two-factor/active`, { data: { code } })
       .then((data: any) => {
@@ -63,11 +64,14 @@ const SecuritySettings = ({ user, userChange }: any) => {
           setTwoFactoring(false);
           setTwoFactorActived(true);
           userChange();
+        } else {
+          message.success('验证失败');
         }
       })
       .catch((error: any) => {
         console.log(error);
-      });
+      })
+      .finally(() => setLoading(false));
   };
 
   const getTwoFactorInfo = () => {
@@ -132,7 +136,7 @@ const SecuritySettings = ({ user, userChange }: any) => {
             onChange={(e) => setCode(e.target.value)}
             placeholder="123456"
           />
-          <Button type="primary" onClick={completeTowFactor}>
+          <Button type="primary" loading={loading} onClick={completeTowFactor}>
             完成设置
           </Button>
         </div>