fix:修复生成的 token 缺失 id导致登录校验出错 (#378)

Co-authored-by: zhongbo <zb123456789>
didi · Aug 4, 2024 · c2ccdca · c2ccdca
1 parent c52a6dd
commit c2ccdca
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/survey-core/src/main/java/com/xiaojusurvey/engine/core/auth/impl/AuthServiceImpl.java b/survey-core/src/main/java/com/xiaojusurvey/engine/core/auth/impl/AuthServiceImpl.java
@@ -63,6 +63,7 @@ public UserVo register(UserParam userParam) {
         user.setUsername(userParam.getUsername());
         user.setPassword(AuthUtil.hash256(userParam.getPassword()));
         mongoRepository.save(user);
+        userParam.setId(user.getId());
         return createTokenAndDeleteCaptcha(userParam);
     }
 
@@ -88,7 +89,8 @@ public UserVo login(UserParam userParam) {
         //验证码
         checkCaptchaIsCorrect(userParam.getCaptchaId(), userParam.getCaptcha());
         //用户验证
-        userService.loadUserByUsernameAndPassword(userParam.getUsername(), userParam.getPassword());
+        User user = userService.loadUserByUsernameAndPassword(userParam.getUsername(), userParam.getPassword());
+        userParam.setId(user.getId());
         //生成token
         return createTokenAndDeleteCaptcha(userParam);
     }