feat(sj_1.2.0-beta2): 增加任务任务类型MapReduce校验

This commit is contained in:
dhb52 2024-09-13 14:43:49 +08:00
parent 99d42cdf9b
commit 7325636d8f

View File

@ -1,5 +1,6 @@
package com.aizuda.snailjob.client.job.core.executor;
import cn.hutool.core.lang.Assert;
import com.aizuda.snailjob.client.job.core.dto.JobArgs;
import com.aizuda.snailjob.client.job.core.dto.MergeReduceArgs;
import com.aizuda.snailjob.client.job.core.dto.ReduceArgs;
@ -18,6 +19,7 @@ public abstract class AbstractMapReduceExecutor extends AbstractMapExecutor {
@Override
public ExecuteResult doJobExecute(final JobArgs jobArgs) {
JobContext jobContext = JobContextManager.getJobContext();
Assert.notNull(jobContext.getMrStage(), "请确认服务器当前定时任务的[任务类型]为MapReduce");
if (jobContext.getMrStage().equals(MapReduceStageEnum.MAP.getStage())) {
return super.doJobExecute(jobArgs);
} else if (jobContext.getMrStage().equals(MapReduceStageEnum.REDUCE.getStage())) {