From ac4ee525d2eab22dd92e900ca706b649ab249329 Mon Sep 17 00:00:00 2001
From: csyue <1203338439@qq.com>
Date: Tue, 28 May 2024 18:31:22 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95=E9=97=AE=E9=A2=98=E9=A1=B9?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../sgzb/auth/controller/TokenController.java  | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/sgzb-auth/src/main/java/com/bonus/sgzb/auth/controller/TokenController.java b/sgzb-auth/src/main/java/com/bonus/sgzb/auth/controller/TokenController.java
index 2e6225c4..d3ff12ab 100644
--- a/sgzb-auth/src/main/java/com/bonus/sgzb/auth/controller/TokenController.java
+++ b/sgzb-auth/src/main/java/com/bonus/sgzb/auth/controller/TokenController.java
@@ -58,12 +58,24 @@ public class TokenController {
     //web端登录
     @PostMapping("login")
     public R<?> login(@RequestBody LoginBody form) throws Exception {
+        // 用户登录
         LoginUser userInfo = sysLoginService.login(form.getUsername(), form.getPassword());
-        if (userInfo != null) {
-            userInfo.setLoginMethod("mobile");
+        String uuid = form.getUuid();
+        String captcha = redisService.getCacheObject(CacheConstants.CAPTCHA_CODE_KEY + uuid).toString();
+        if (StringUtils.isBlank(captcha)) {
+            // 删除验证码缓存
+            redisService.deleteObject(CacheConstants.CAPTCHA_CODE_KEY + uuid);
+            return R.fail("验证码超时，请重新刷新");
+        }
+        if (form.getCode() != null && form.getCode().equals(captcha)) {
+            // 删除验证码缓存
+            redisService.deleteObject(CacheConstants.CAPTCHA_CODE_KEY + uuid);
+            // 获取登录token
             return R.ok(tokenService.createToken(userInfo));
         } else {
-            return R.fail("登录信息为空，请重试");
+            // 删除验证码缓存
+            redisService.deleteObject(CacheConstants.CAPTCHA_CODE_KEY + uuid);
+            return R.fail("验证码错误");
         }
     }