From e394d65ec0ad59eaed288ee274ea438c63bac13e Mon Sep 17 00:00:00 2001 From: opensnail <598092184@qq.com> Date: Mon, 17 Jun 2024 18:32:04 +0800 Subject: [PATCH] =?UTF-8?q?fix(sj=5F1.1.0):=20=E6=94=AF=E6=8C=81=E5=B7=A5?= =?UTF-8?q?=E4=BD=9C=E6=B5=81=E5=85=A8=E5=B1=80=E4=B8=8A=E4=B8=8B=E6=96=87?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../job/core/executor/AbstractMapReduceExecutor.java | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/snail-job-client/snail-job-client-job-core/src/main/java/com/aizuda/snailjob/client/job/core/executor/AbstractMapReduceExecutor.java b/snail-job-client/snail-job-client-job-core/src/main/java/com/aizuda/snailjob/client/job/core/executor/AbstractMapReduceExecutor.java index 0c801e23..4a59f00f 100644 --- a/snail-job-client/snail-job-client-job-core/src/main/java/com/aizuda/snailjob/client/job/core/executor/AbstractMapReduceExecutor.java +++ b/snail-job-client/snail-job-client-job-core/src/main/java/com/aizuda/snailjob/client/job/core/executor/AbstractMapReduceExecutor.java @@ -22,14 +22,19 @@ public abstract class AbstractMapReduceExecutor extends AbstractMapExecutor { public ExecuteResult doJobExecute(final JobArgs jobArgs) { JobContext jobContext = JobContextManager.getJobContext(); if (jobContext.getMrStage().equals(MapReduceStageEnum.MAP.getStage())) { - return super.doJobExecute(jobArgs); - } else if(jobContext.getMrStage().equals(MapReduceStageEnum.REDUCE.getStage())) { + return super.doJobExecute(jobArgs); + } else if (jobContext.getMrStage().equals(MapReduceStageEnum.REDUCE.getStage())) { ReduceArgs reduceArgs = (ReduceArgs) jobArgs; - return doReduceExecute(reduceArgs); + return this.doReduceExecute(reduceArgs); + } else if (jobContext.getMrStage().equals(MapReduceStageEnum.MERGE_REDUCE.getStage())) { + ReduceArgs reduceArgs = (ReduceArgs) jobArgs; + return this.doMergeReduceExecute(reduceArgs); } throw new SnailJobMapReduceException("非法的MapReduceStage"); } protected abstract ExecuteResult doReduceExecute(ReduceArgs reduceArgs); + + protected abstract ExecuteResult doMergeReduceExecute(ReduceArgs reduceArgs); }