From a9ebc855851bb4d58ed5c8851e9a8ce7f608f776 Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Thu, 22 Sep 2022 11:42:44 +0800
Subject: [PATCH] 任务计划路线范围添加修改

---
 src/main/java/org/springblade/common/tool/EnhancementServicesTool.java |   21 ++++-----------------
 1 files changed, 4 insertions(+), 17 deletions(-)

diff --git a/src/main/java/org/springblade/common/tool/EnhancementServicesTool.java b/src/main/java/org/springblade/common/tool/EnhancementServicesTool.java
index f98484f..1e08c5d 100644
--- a/src/main/java/org/springblade/common/tool/EnhancementServicesTool.java
+++ b/src/main/java/org/springblade/common/tool/EnhancementServicesTool.java
@@ -3,6 +3,7 @@
 import org.springblade.core.mp.base.BaseEntity;
 import org.springblade.core.secure.BladeUser;
 import org.springblade.core.secure.utils.AuthUtil;
+import org.springblade.core.tenant.mp.TenantEntity;
 import org.springblade.core.tool.utils.*;
 
 import java.lang.reflect.Field;
@@ -17,7 +18,7 @@
  * @Date 2022/9/20 16:16
  * @Version 1.0
  */
-public class EnhancementServicesTool <T extends BaseEntity> {
+public class EnhancementServicesTool <T extends TenantEntity> {
 	public void resolveEntity(T entity) {
 		try {
 			BladeUser user = AuthUtil.getUser();
@@ -40,24 +41,10 @@
 
 			entity.setUpdateTime(now);
 			entity.setIsDeleted(0);
-			Field field = ReflectUtil.getField(entity.getClass(), "tenantId");
-			if (ObjectUtil.isNotEmpty(field)) {
-				Method getTenantId = ClassUtil.getMethod(entity.getClass(), "getTenantId", new Class[0]);
-				String tenantId = String.valueOf(getTenantId.invoke(entity));
-				if (ObjectUtil.isEmpty(tenantId)) {
-					Method setTenantId = ClassUtil.getMethod(entity.getClass(), "setTenantId", new Class[]{String.class});
-					setTenantId.invoke(entity, null);
-				}
-			}
+			entity.setTenantId(user.getTenantId());
 
 		} catch (Throwable var8) {
-			try {
-				throw var8;
-			} catch (IllegalAccessException e) {
-				throw new RuntimeException(e);
-			} catch (InvocationTargetException e) {
-				throw new RuntimeException(e);
-			}
+			throw var8;
 		}
 	}
 }

--
Gitblit v1.9.3