lbqms.git - Gitblit

			@@ -8,7 +8,6 @@
			import lombok.NoArgsConstructor;
			import lombok.extern.slf4j.Slf4j;
			import org.dromara.common.core.constant.GlobalConstants;
			import org.dromara.common.core.context.ThreadLocalHolder;
			import org.dromara.common.core.utils.SpringUtils;
			import org.dromara.common.core.utils.StringUtils;
			import org.dromara.common.redis.utils.RedisUtils;
			@@ -27,7 +26,7 @@

			private static final String DYNAMIC_TENANT_KEY = GlobalConstants.GLOBAL_REDIS_KEY + "dynamicTenant";

			private static final String TENANT_ID_KEY = "tempDynamicTenant";
			private static final ThreadLocal<String> TEMP_DYNAMIC_TENANT = new ThreadLocal<>();

			/**
			* 租户功能是否启用
			@@ -78,22 +77,28 @@
			}
			}

			public static void setDynamic(String tenantId) {
			setDynamic(tenantId, false);
			}

			/**
			* 设置动态租户(一直有效需要手动清理)
			* <p>
			* 如果为未登录状态下那么只在当前线程内生效
			*
			* @param tenantId 租户id
			* @param global 是否全局生效
			*/
			public static void setDynamic(String tenantId) {
			public static void setDynamic(String tenantId, boolean global) {
			if (!isEnable()) {
			return;
			}
			if (!isLogin()) {
			ThreadLocalHolder.set(TENANT_ID_KEY, tenantId);
			if (!isLogin() \|\| !global) {
			TEMP_DYNAMIC_TENANT.set(tenantId);
			return;
			}
			String cacheKey = DYNAMIC_TENANT_KEY + ":" + LoginHelper.getUserId();
			RedisUtils.setCacheObject(cacheKey, tenantId);
			ThreadLocalHolder.set(cacheKey, tenantId);
			}

			/**
			@@ -106,15 +111,15 @@
			return null;
			}
			if (!isLogin()) {
			return ThreadLocalHolder.get(TENANT_ID_KEY);
			return TEMP_DYNAMIC_TENANT.get();
			}
			String cacheKey = DYNAMIC_TENANT_KEY + ":" + LoginHelper.getUserId();
			String tenantId = ThreadLocalHolder.get(cacheKey);
			// 如果线程内有值优先返回
			String tenantId = TEMP_DYNAMIC_TENANT.get();
			if (StringUtils.isNotBlank(tenantId)) {
			return tenantId;
			}
			String cacheKey = DYNAMIC_TENANT_KEY + ":" + LoginHelper.getUserId();
			tenantId = RedisUtils.getCacheObject(cacheKey);
			ThreadLocalHolder.set(cacheKey, tenantId);
			return tenantId;
			}

			@@ -126,12 +131,12 @@
			return;
			}
			if (!isLogin()) {
			ThreadLocalHolder.remove(TENANT_ID_KEY);
			TEMP_DYNAMIC_TENANT.remove();
			return;
			}
			TEMP_DYNAMIC_TENANT.remove();
			String cacheKey = DYNAMIC_TENANT_KEY + ":" + LoginHelper.getUserId();
			RedisUtils.deleteObject(cacheKey);
			ThreadLocalHolder.remove(cacheKey);
			}

			/**