From dd611ad3fcf9c3ad26616fc151c608259b69eb67 Mon Sep 17 00:00:00 2001
From: FatttSnake <fatttsnake@gmail.com>
Date: Thu, 18 May 2023 17:03:02 +0800
Subject: [PATCH] Optimized the front-end login logic

---
 ui/src/pages/Login.vue | 42 ++++++++++++++++++++++++++----------------
 1 file changed, 26 insertions(+), 16 deletions(-)

diff --git a/ui/src/pages/Login.vue b/ui/src/pages/Login.vue
index 767a4dd..a0e9574 100644
--- a/ui/src/pages/Login.vue
+++ b/ui/src/pages/Login.vue
@@ -84,7 +84,7 @@
 import { getCaptchaSrc, login, verifyCaptcha } from '@/utils/auth'
 import backShape from '@/assets/svg/back-shape.svg'
 import { ElMessage } from 'element-plus'
-import { LOGIN_SUCCESS, PRODUCTION_NAME } from '@/constants/Common.constants'
+import { LOGIN_SUCCESS, LOGOUT_FAILED, PRODUCTION_NAME } from '@/constants/Common.constants'
 import { setToken } from '@/utils/common'
 
 export default {
@@ -151,21 +151,31 @@ export default {
             }
             login(this.userName, this.password).then((res) => {
                 const data = res.data
-                if (data.code === LOGIN_SUCCESS) {
-                    setToken(data.data.token)
-                    ElMessage.success({
-                        dangerouslyUseHTMLString: true,
-                        message: '<strong>登录成功</strong>'
-                    })
-                    setTimeout(function () {
-                        _this.$router.push('/')
-                    }, 1500)
-                } else {
-                    ElMessage.error({
-                        dangerouslyUseHTMLString: true,
-                        message: '<strong>用户名</strong> 或 <strong>密码</strong> 错误'
-                    })
-                    this.resetLogin()
+                switch (data.code) {
+                    case LOGIN_SUCCESS:
+                        setToken(data.data.token)
+                        ElMessage.success({
+                            dangerouslyUseHTMLString: true,
+                            message: '<strong>登录成功</strong>'
+                        })
+                        setTimeout(function () {
+                            _this.$router.push('/')
+                        }, 1500)
+                        break
+                    case LOGOUT_FAILED:
+                        ElMessage.error({
+                            dangerouslyUseHTMLString: true,
+                            message: '<strong>用户名</strong> 或 <strong>密码</strong> 错误'
+                        })
+                        this.resetLogin()
+                        break
+                    default:
+                        ElMessage.error({
+                            dangerouslyUseHTMLString: true,
+                            message: '<strong>服务器出错了</strong>，请稍后重试'
+                        })
+                        this.resetLogin()
+                        break
                 }
             })
         }