diff --git a/src/main/java/com/chinaweal/aiccs/org/controller/OAuth2Controller.java b/src/main/java/com/chinaweal/aiccs/org/controller/OAuth2Controller.java index cec1a2d..0e539a7 100644 --- a/src/main/java/com/chinaweal/aiccs/org/controller/OAuth2Controller.java +++ b/src/main/java/com/chinaweal/aiccs/org/controller/OAuth2Controller.java @@ -749,6 +749,9 @@ public class OAuth2Controller extends BaseController { request.getSession().setAttribute(DRUID_SESSION_KEY, String.format("%s(%s)", user.getName(), user.getEname())); redisService.remove(tuser.getUsername()); SSOUtil.login(user); + if (StringUtils.notEquals(code, "test")) { + redisService.remove(USER_LOGIN_SELECTION + code); + } return RestResult.ok(); }