Bladeren bron

检查定时任务bean所在包名是否为白名单配置

RuoYi 4 jaren geleden
bovenliggende
commit
4df2b72b30
1 gewijzigde bestanden met toevoegingen van 4 en 2 verwijderingen
  1. 4 2
      ruoyi-modules/ruoyi-job/src/main/java/com/ruoyi/job/util/ScheduleUtils.java

+ 4 - 2
ruoyi-modules/ruoyi-job/src/main/java/com/ruoyi/job/util/ScheduleUtils.java

@@ -14,6 +14,7 @@ import com.ruoyi.common.core.constant.Constants;
14 14
 import com.ruoyi.common.core.constant.ScheduleConstants;
15 15
 import com.ruoyi.common.core.exception.job.TaskException;
16 16
 import com.ruoyi.common.core.exception.job.TaskException.Code;
17
+import com.ruoyi.common.core.utils.SpringUtils;
17 18
 import com.ruoyi.common.core.utils.StringUtils;
18 19
 import com.ruoyi.job.domain.SysJob;
19 20
 
@@ -127,6 +128,7 @@ public class ScheduleUtils
127 128
         {
128 129
             return StringUtils.containsAnyIgnoreCase(invokeTarget, Constants.JOB_WHITELIST_STR);
129 130
         }
130
-        return true;
131
+        Object obj = SpringUtils.getBean(StringUtils.split(invokeTarget, ".")[0]);
132
+        return StringUtils.containsAnyIgnoreCase(obj.getClass().getPackage().getName(), Constants.JOB_WHITELIST_STR);
131 133
     }
132
-}
134
+}