SaTokenDao 新增默认方法 onChange，当 SaManager.saTokenDao 变更时，调用该方法

dromara · Oct 14, 2023 · 8db78e0 · 8db78e0
1 parent f2416a6
commit 8db78e0
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 7 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
@@ -102,8 +102,8 @@ public static void setSaTokenDao(SaTokenDao saTokenDao) {
 		SaTokenEventCenter.doRegisterComponent("SaTokenDao", saTokenDao);
 	}
 	private static void setSaTokenDaoMethod(SaTokenDao saTokenDao) {
-		if((SaManager.saTokenDao instanceof SaTokenDaoDefaultImpl)) {
-			((SaTokenDaoDefaultImpl)SaManager.saTokenDao).endRefreshThread();
+		if (SaManager.saTokenDao != null) {
+			SaManager.saTokenDao.onChange();
 		}
 		SaManager.saTokenDao = saTokenDao;
 	}

diff --git a/sa-token-core/src/main/java/cn/dev33/satoken/dao/SaTokenDao.java b/sa-token-core/src/main/java/cn/dev33/satoken/dao/SaTokenDao.java
@@ -199,6 +199,13 @@ default void updateSessionTimeout(String sessionId, long timeout) {
 	 * @return 查询到的数据集合 
 	 */
 	List<String> searchData(String prefix, String keyword, int start, int size, boolean sortType);
-
-
+
+
+	// --------------------- 实例相关 ---------------------
+
+	/**
+	 * 当 SaManager.saTokenDao 变更时，调用该方法
+	 */
+	default void onChange() {
+	}
 }
diff --git a/sa-token-core/src/main/java/cn/dev33/satoken/dao/SaTokenDaoDefaultImpl.java b/sa-token-core/src/main/java/cn/dev33/satoken/dao/SaTokenDaoDefaultImpl.java
@@ -260,12 +260,12 @@ public void initRefreshThread() {
 		});
 		this.refreshThread.start();
 	}
-	
+
 	/**
 	 * 结束定时任务，不再定时清理过期数据
 	 */
-	public void endRefreshThread() {
+	@Override
+	public void onChange() {
 		this.refreshFlag = false;
 	}
-
 }