From c26adb716a03c9fb0ed21e3caf6abacfadf3680b Mon Sep 17 00:00:00 2001 From: click33 <2393584716@qq.com> Date: Thu, 27 Oct 2022 11:59:49 +0800 Subject: [PATCH] =?UTF-8?q?`SaManager.getStpLogic("type")`=20=E9=BB=98?= =?UTF-8?q?=E8=AE=A4=E5=BD=93=E5=AF=B9=E5=BA=94type=E4=B8=8D=E5=AD=98?= =?UTF-8?q?=E5=9C=A8=E6=97=B6=E4=B8=8D=E5=86=8D=E6=8A=9B=E5=87=BA=E5=BC=82?= =?UTF-8?q?=E5=B8=B8=EF=BC=8C=E8=80=8C=E6=98=AF=E8=87=AA=E5=8A=A8=E5=88=9B?= =?UTF-8?q?=E5=BB=BA=E5=B9=B6=E8=BF=94=E5=9B=9E=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../main/java/cn/dev33/satoken/SaManager.java | 52 ++++++++++++++----- .../cn/dev33/satoken/strategy/SaStrategy.java | 10 ++-- sa-token-doc/more/common-action.md | 3 +- 3 files changed, 45 insertions(+), 20 deletions(-) diff --git a/sa-token-core/src/main/java/cn/dev33/satoken/SaManager.java b/sa-token-core/src/main/java/cn/dev33/satoken/SaManager.java index d0c56888..8aadf8e4 100644 --- a/sa-token-core/src/main/java/cn/dev33/satoken/SaManager.java +++ b/sa-token-core/src/main/java/cn/dev33/satoken/SaManager.java @@ -1,6 +1,6 @@ package cn.dev33.satoken; -import java.util.HashMap; +import java.util.LinkedHashMap; import java.util.Map; import cn.dev33.satoken.config.SaTokenConfig; @@ -214,7 +214,7 @@ public class SaManager { /** * StpLogic集合, 记录框架所有成功初始化的StpLogic */ - public static Map stpLogicMap = new HashMap(); + public static Map stpLogicMap = new LinkedHashMap(); /** * 向全局集合中 put 一个 StpLogic @@ -225,32 +225,56 @@ public class SaManager { } /** - * 根据 LoginType 获取对应的StpLogic，如果不存在则抛出异常 + * 根据 LoginType 获取对应的StpLogic，如果不存在则新建并返回 * @param loginType 对应的账号类型 * @return 对应的StpLogic */ public static StpLogic getStpLogic(String loginType) { - // 如果type为空则返回框架内置的 + return getStpLogic(loginType, true); + } + + /** + * 根据 LoginType 获取对应的StpLogic，如果不存在，isCreate参数=是否自动创建并返回 + * @param loginType 对应的账号类型 + * @param isCreate 在 StpLogic 不存在时，true=新建并返回，false=抛出异常 + * @return 对应的StpLogic + */ + public static StpLogic getStpLogic(String loginType, boolean isCreate) { + // 如果type为空则返回框架默认内置的 if(loginType == null || loginType.isEmpty()) { return StpUtil.stpLogic; } - // 从 stpLogicMap 中获取 + // 从集合中获取 StpLogic stpLogic = stpLogicMap.get(loginType); if(stpLogic == null) { - /* - * 此时有两种情况会造成 StpLogic == null - * 1. loginType拼写错误，请改正 （建议使用常量） - * 2. 自定义StpUtil尚未初始化（静态类中的属性至少一次调用后才会初始化），解决方法两种 - * (1) 从main方法里调用一次 - * (2) 在自定义StpUtil类加上类似 @Component 的注解让容器启动时扫描到自动初始化 - */ - throw new SaTokenException("未能获取对应StpLogic，type="+ loginType); + + // isCreate=true时，自创建模式：自动创建并返回 + if(isCreate) { + synchronized (SaManager.class) { + stpLogic = stpLogicMap.get(loginType); + if(stpLogic == null) { + stpLogic = new StpLogic(loginType); + // 此处无需手动put，因为 StpLogic 构造方法中会自动put + // putStpLogic(stpLogic); + } + } + } + // isCreate=false时，严格校验模式：抛出异常 + else { + /* + * 此时有两种情况会造成 StpLogic == null + * 1. loginType拼写错误，请改正 （建议使用常量） + * 2. 自定义StpUtil尚未初始化（静态类中的属性至少一次调用后才会初始化），解决方法两种 + * (1) 从main方法里调用一次 + * (2) 在自定义StpUtil类加上类似 @Component 的注解让容器启动时扫描到自动初始化 + */ + throw new SaTokenException("未能获取对应StpLogic，type="+ loginType); + } } // 返回 return stpLogic; } - } diff --git a/sa-token-core/src/main/java/cn/dev33/satoken/strategy/SaStrategy.java b/sa-token-core/src/main/java/cn/dev33/satoken/strategy/SaStrategy.java index a012cc54..56186e97 100644 --- a/sa-token-core/src/main/java/cn/dev33/satoken/strategy/SaStrategy.java +++ b/sa-token-core/src/main/java/cn/dev33/satoken/strategy/SaStrategy.java @@ -142,31 +142,31 @@ public final class SaStrategy { // 校验 @SaCheckLogin 注解 SaCheckLogin checkLogin = (SaCheckLogin) SaStrategy.me.getAnnotation.apply(target, SaCheckLogin.class); if(checkLogin != null) { - SaManager.getStpLogic(checkLogin.type()).checkByAnnotation(checkLogin); + SaManager.getStpLogic(checkLogin.type(), false).checkByAnnotation(checkLogin); } // 校验 @SaCheckRole 注解 SaCheckRole checkRole = (SaCheckRole) SaStrategy.me.getAnnotation.apply(target, SaCheckRole.class); if(checkRole != null) { - SaManager.getStpLogic(checkRole.type()).checkByAnnotation(checkRole); + SaManager.getStpLogic(checkRole.type(), false).checkByAnnotation(checkRole); } // 校验 @SaCheckPermission 注解 SaCheckPermission checkPermission = (SaCheckPermission) SaStrategy.me.getAnnotation.apply(target, SaCheckPermission.class); if(checkPermission != null) { - SaManager.getStpLogic(checkPermission.type()).checkByAnnotation(checkPermission); + SaManager.getStpLogic(checkPermission.type(), false).checkByAnnotation(checkPermission); } // 校验 @SaCheckSafe 注解 SaCheckSafe checkSafe = (SaCheckSafe) SaStrategy.me.getAnnotation.apply(target, SaCheckSafe.class); if(checkSafe != null) { - SaManager.getStpLogic(checkSafe.type()).checkByAnnotation(checkSafe); + SaManager.getStpLogic(checkSafe.type(), false).checkByAnnotation(checkSafe); } // 校验 @SaCheckDisable 注解 SaCheckDisable checkDisable = (SaCheckDisable) SaStrategy.me.getAnnotation.apply(target, SaCheckDisable.class); if(checkDisable != null) { - SaManager.getStpLogic(checkDisable.type()).checkByAnnotation(checkDisable); + SaManager.getStpLogic(checkDisable.type(), false).checkByAnnotation(checkDisable); } // 校验 @SaCheckBasic 注解 diff --git a/sa-token-doc/more/common-action.md b/sa-token-doc/more/common-action.md index c4b6da58..504a160d 100644 --- a/sa-token-doc/more/common-action.md +++ b/sa-token-doc/more/common-action.md @@ -16,7 +16,8 @@ SaManager.getSaTokenListener(); // 获取侦听器对象 SaManager.getSaTemp(); // 获取临时令牌认证模块对象 SaManager.getSaJsonTemplate(); // 获取 JSON 转换器 Bean SaManager.getSaSignTemplate(); // 获取参数签名 Bean -SaManager.getStpLogic("type"); // 获取指定账号类型的StpLogic对象 +SaManager.getStpLogic("type"); // 获取指定账号类型的StpLogic对象，获取不到时自动创建并返回 +SaManager.getStpLogic("type", false); // 获取指定账号类型的StpLogic对象，获取不到时抛出异常 SaManager.putStpLogic(stpLogic); // 向全局集合中 put 一个 StpLogic ```