gtsoft/gtsoft-snail-job-demo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增MapReduce测试类

Browse Source
This commit is contained in:
zhengweilin 2024-06-16 21:31:27 +08:00
parent e05bc9067f
commit b72899eafa
2 changed files with 41 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/main/java/com/example/snailjob/job/MapReduceJobExecutor.java
View File

@ -1,36 +0,0 @@
package com.example.snailjob.job;
import com.aizuda.snailjob.client.job.core.dto.JobArgs;
import com.aizuda.snailjob.client.job.core.dto.MapReduceArgs;
import com.aizuda.snailjob.client.job.core.executor.AbstractMapReduceExecutor;
import com.aizuda.snailjob.client.model.ExecuteResult;
import com.aizuda.snailjob.common.core.constant.SystemConstants;
import com.aizuda.snailjob.common.core.model.JobContext;
import com.google.common.collect.Lists;
import org.springframework.stereotype.Component;
/**
* @author: opensnail
* @date : 2024-06-13
* @since : sj_1.1.0
*/
@Component
public class MapReduceJobExecutor extends AbstractMapReduceExecutor {
@Override
public ExecuteResult doJobExecute(final MapReduceArgs mapReduceArgs) {
if (mapReduceArgs.getMapName().equals(SystemConstants.MAP_ROOT)) {
doMapExecute(Lists.newArrayList(Lists.newArrayList(1, 2, 3), Lists.newArrayList(4, 5, 6)),
"TWO_MAP");
return ExecuteResult.success();
} else {
return ExecuteResult.success(1);
}
}
@Override
protected ExecuteResult doReduceExecute(final JobContext jobContext, final JobArgs jobArgs) {
return null;
}
}

41
src/main/java/com/example/snailjob/job/TestMapReduceJobExecutor.java Normal file
View File

@ -0,0 +1,41 @@
package com.example.snailjob.job;
import com.aizuda.snailjob.client.job.core.dto.MapArgs;
import com.aizuda.snailjob.client.job.core.dto.ReduceArgs;
import com.aizuda.snailjob.client.job.core.executor.AbstractMapReduceExecutor;
import com.aizuda.snailjob.client.model.ExecuteResult;
import com.aizuda.snailjob.common.core.constant.SystemConstants;
import com.google.common.collect.Lists;
import org.springframework.stereotype.Component;
/**
* @author: opensnail
* @date : 2024-06-13
* @since : sj_1.1.0
*/
@Component
public class TestMapReduceJobExecutor extends AbstractMapReduceExecutor {
@Override
public ExecuteResult doJobMapExecute(MapArgs mapArgs) {
if (mapArgs.getTaskName().equals(SystemConstants.MAP_ROOT)) {
doMap(Lists.newArrayList(Lists.newArrayList(1, 2, 3), Lists.newArrayList(4, 5, 6)), "TWO_MAP");
return ExecuteResult.success();
} else if (mapArgs.getTaskName().equals("TWO_MAP")) {
return ExecuteResult.success(mapArgs.getArgsStr());
} else {
return ExecuteResult.success();
}
}
@Override
protected ExecuteResult doReduceExecute(ReduceArgs reduceArgs) {
return ExecuteResult.success(reduceArgs.getMapResult());
}
@Override
protected ExecuteResult doMergeReduceExecute(ReduceArgs reduceArgs) {
return ExecuteResult.success(reduceArgs.getMapResult());
}
}