Просмотр исходного кода

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

RuoYi лет назад: 4
Родитель
Сommit
4df2b72b30
1 измененных файлов с 4 добавлено и 2 удалено
  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
 import com.ruoyi.common.core.constant.ScheduleConstants;
14
 import com.ruoyi.common.core.constant.ScheduleConstants;
15
 import com.ruoyi.common.core.exception.job.TaskException;
15
 import com.ruoyi.common.core.exception.job.TaskException;
16
 import com.ruoyi.common.core.exception.job.TaskException.Code;
16
 import com.ruoyi.common.core.exception.job.TaskException.Code;
17
+import com.ruoyi.common.core.utils.SpringUtils;
17
 import com.ruoyi.common.core.utils.StringUtils;
18
 import com.ruoyi.common.core.utils.StringUtils;
18
 import com.ruoyi.job.domain.SysJob;
19
 import com.ruoyi.job.domain.SysJob;
19
 
20
 
@@ -127,6 +128,7 @@ public class ScheduleUtils
127
         {
128
         {
128
             return StringUtils.containsAnyIgnoreCase(invokeTarget, Constants.JOB_WHITELIST_STR);
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
+}