From 0f1cfc129ccab75b1cedec8a618865db71cfa973 Mon Sep 17 00:00:00 2001 From: Chopper Date: Thu, 13 Oct 2022 14:32:40 +0800 Subject: [PATCH] =?UTF-8?q?keys=E5=88=A4=E5=AE=9A=E4=BC=98=E5=8C=96?= =?UTF-8?q?=E4=B8=BA=E7=B2=BE=E5=87=86=E5=88=A4=E5=AE=9A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../java/cn/lili/common/security/context/UserContext.java | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/framework/src/main/java/cn/lili/common/security/context/UserContext.java b/framework/src/main/java/cn/lili/common/security/context/UserContext.java index a9b69d31..78d3e5a6 100644 --- a/framework/src/main/java/cn/lili/common/security/context/UserContext.java +++ b/framework/src/main/java/cn/lili/common/security/context/UserContext.java @@ -1,6 +1,7 @@ package cn.lili.common.security.context; import cn.lili.cache.Cache; +import cn.lili.cache.CachePrefix; import cn.lili.common.enums.ResultCode; import cn.lili.common.exception.ServiceException; import cn.lili.common.security.AuthUser; @@ -13,7 +14,6 @@ import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.ServletRequestAttributes; import javax.servlet.http.HttpServletRequest; -import java.util.Objects; /** * 用户上下文 @@ -61,11 +61,13 @@ public class UserContext { */ public static AuthUser getAuthUser(Cache cache, String accessToken) { try { + AuthUser authUser = getAuthUser(accessToken); + assert authUser != null; - if (!cache.hasKey("*" + Objects.requireNonNull(UserContext.getAuthUser(accessToken)).getRole().name() + accessToken)) { + if (!cache.hasKey(CachePrefix.ACCESS_TOKEN.getPrefix(authUser.getRole()) + accessToken)) { throw new ServiceException(ResultCode.USER_AUTHORITY_ERROR); } - return getAuthUser(accessToken); + return authUser; } catch (Exception e) { return null; }