From 031d1c5a7c680bfcf2b9a4a98e9d0ae0c613ff51 Mon Sep 17 00:00:00 2001 From: zhuo <2089219845@qq.com> Date: 星期六, 15 三月 2025 11:42:24 +0800 Subject: [PATCH] 新增无需密码获取用户登录信息 --- ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysLoginService.java | 56 ++++++++++++++++++++++++++++++++++++++++++++++++-------- 1 files changed, 48 insertions(+), 8 deletions(-) diff --git a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysLoginService.java b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysLoginService.java index b6841dd..c5da313 100644 --- a/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysLoginService.java +++ b/ruoyi-framework/src/main/java/com/ruoyi/framework/web/service/SysLoginService.java @@ -9,6 +9,7 @@ import com.ruoyi.common.core.domain.entity.User; import com.ruoyi.framework.model.SsoOauthTokenModel; import com.ruoyi.framework.model.SsoUserInfoModel; +import com.ruoyi.framework.web.ssoAuth.SsoCodeAuthenticationToken; import com.ruoyi.system.mapper.UserMapper; import lombok.extern.slf4j.Slf4j; import org.apache.http.HttpEntity; @@ -93,9 +94,6 @@ @Autowired private RedisTemplate redisTemplate; - - @Autowired - private UserMapper userMapper; /** * 鐧诲綍楠岃瘉 @@ -234,26 +232,68 @@ // 鑾峰彇鍗曠偣鐧诲綍token SsoOauthTokenModel tokenModel = this.getSsoAccessToken(code); if (tokenModel == null) { - return null; + throw new RuntimeException("鍗曠偣鐧诲綍楠岃瘉澶辫触"); } // 鑾峰彇鍗曠偣鐧诲綍鐢ㄦ埛淇℃伅 SsoUserInfoModel userInfo = this.getSsoUserInfo(tokenModel.getAccess_token()); if (userInfo == null) { - return null; + throw new RuntimeException("鍗曠偣鐧诲綍楠岃瘉澶辫触"); } // 鏌ヨ鏈湴鐢ㄦ埛淇℃伅 + LoginUser loginUser = this.getSsoLoginToken(userInfo.getEmployeeId()); + recordLoginInfo(loginUser.getUserId()); - - + // 鍒涘缓鐧诲綍淇℃伅 AsyncManager.me().execute(AsyncFactory.recordLogininfor(userInfo.getEmployeeId(), Constants.LOGIN_SUCCESS, MessageUtils.message("user.single.login.success"))); redisTemplate.opsForValue().set("ssoOauthToken:idToken:" + userInfo.getEmployeeId(), tokenModel.getId_token(), 3, TimeUnit.DAYS); // 鐢熸垚token - return tokenService.createToken(null); + return tokenService.createToken(loginUser); } /** + * ***** 鑾峰彇鏈湴鐢ㄦ埛淇℃伅 ***** + * @param username + * @return + */ + public LoginUser getSsoLoginToken(String username) { + // 鐢ㄦ埛楠岃瘉 + Authentication authentication = null; + SysUser sysUser = sysUserService.selectUserByUserName(username); + if (sysUser == null) { + throw new ServiceException("鐢ㄦ埛涓嶅瓨鍦�"); + } + try + { + // 鏃犻渶瀵嗙爜鑾峰彇鐢ㄦ埛淇℃伅 + authentication = authenticationManager.authenticate(new SsoCodeAuthenticationToken(username)); + } + catch (Exception e) + { + if (e instanceof BadCredentialsException) + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.password.not.match"))); + throw new UserPasswordNotMatchException(); + } + else + { + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, e.getMessage())); + throw new ServiceException(e.getMessage()); + } + } + finally + { + AuthenticationContextHolder.clearContext(); + } + AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_SUCCESS, MessageUtils.message("user.login.success"))); + return (LoginUser) authentication.getPrincipal(); + + } + + + + /** * **** 鑾峰彇鍗曠偣鐧诲綍token **** * @param code * @return -- Gitblit v1.9.3