guoenzhou 1 year ago
parent
commit
45af4cdcea

+ 29 - 0
base-common/ruoyi-common-datascope/src/main/java/com/ruoyi/common/datascope/context/DataScopeContextHolder.java

@@ -0,0 +1,29 @@
+package com.ruoyi.common.datasource.context;
+
+/**
+ * DataScope上下文 Holder
+ *
+ * @author yq
+ */
+public class DataScopeContextHolder {
+
+    /**
+     * 当前租户编号
+     */
+ private static final ThreadLocal<String> DATA_SCOPE_SQL = new ThreadLocal<>();
+
+
+    public static String getDataScopeSql() {
+        return DATA_SCOPE_SQL.get();
+    }
+
+
+    public static void setTenantId(String dataScope) {
+        DATA_SCOPE_SQL.set(dataScope);
+    }
+
+    public static void clear() {
+        DATA_SCOPE_SQL.remove();
+    }
+
+}