From 95c01301f65379e7634e4619bf4c49186aa5be41 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子Li <15040126243@163.com>
Date: 星期五, 07 二月 2025 14:19:28 +0800
Subject: [PATCH] !644 同步修复一些问题 Merge pull request !644 from 疯狂的狮子Li/dev

---
 ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/helper/DataPermissionHelper.java |   30 ++++++++++++++++++++++++++++--
 1 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/helper/DataPermissionHelper.java b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/helper/DataPermissionHelper.java
index ccb0fca..f03d74e 100644
--- a/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/helper/DataPermissionHelper.java
+++ b/ruoyi-common/ruoyi-common-mybatis/src/main/java/org/dromara/common/mybatis/helper/DataPermissionHelper.java
@@ -9,6 +9,7 @@
 import lombok.AccessLevel;
 import lombok.NoArgsConstructor;
 import org.dromara.common.core.utils.reflect.ReflectUtils;
+import org.dromara.common.mybatis.annotation.DataPermission;
 
 import java.util.HashMap;
 import java.util.Map;
@@ -28,6 +29,33 @@
     private static final String DATA_PERMISSION_KEY = "data:permission";
 
     private static final ThreadLocal<Stack<Integer>> REENTRANT_IGNORE = ThreadLocal.withInitial(Stack::new);
+
+    private static final ThreadLocal<DataPermission> PERMISSION_CACHE = new ThreadLocal<>();
+
+    /**
+     * 鑾峰彇褰撳墠鎵цmapper鏉冮檺娉ㄨВ
+     *
+     * @return 杩斿洖褰撳墠鎵цmapper鏉冮檺娉ㄨВ
+     */
+    public static DataPermission getPermission() {
+        return PERMISSION_CACHE.get();
+    }
+
+    /**
+     * 璁剧疆褰撳墠鎵цmapper鏉冮檺娉ㄨВ
+     *
+     * @param dataPermission   鏁版嵁鏉冮檺娉ㄨВ
+     */
+    public static void setPermission(DataPermission dataPermission) {
+        PERMISSION_CACHE.set(dataPermission);
+    }
+
+    /**
+     * 鍒犻櫎褰撳墠鎵цmapper鏉冮檺娉ㄨВ
+     */
+    public static void removePermission() {
+        PERMISSION_CACHE.remove();
+    }
 
     /**
      * 浠庝笂涓嬫枃涓幏鍙栨寚瀹氶敭鐨勫彉閲忓�硷紝骞跺皢鍏惰浆鎹负鎸囧畾鐨勭被鍨�
@@ -119,7 +147,6 @@
 
     /**
      * 鍦ㄥ拷鐣ユ暟鎹潈闄愪腑鎵ц
-     * <p>绂佹鍦ㄥ拷鐣ユ暟鎹潈闄愪腑鎵ц蹇界暐鏁版嵁鏉冮檺</p>
      *
      * @param handle 澶勭悊鎵ц鏂规硶
      */
@@ -134,7 +161,6 @@
 
     /**
      * 鍦ㄥ拷鐣ユ暟鎹潈闄愪腑鎵ц
-     * <p>绂佹鍦ㄥ拷鐣ユ暟鎹潈闄愪腑鎵ц蹇界暐鏁版嵁鏉冮檺</p>
      *
      * @param handle 澶勭悊鎵ц鏂规硶
      */

--
Gitblit v1.9.3