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

优化Map测试用例

Browse Source
This commit is contained in:
byteblogs168 2024-06-27 11:02:53 +08:00
parent 550e263d99
commit b170277f0f
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java
View File

@ -27,27 +27,24 @@ public class TestAnnoMapReduceJobExecutor {
@MapExecutor
public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
System.out.println(mapArgs);
return mapHandler.doMap(Lists.newArrayList(1, 2, 3, 4, 5, 6), "MONTH_MAP");
return mapHandler.doMap(Lists.newArrayList("1", "2", "3", "4", "5", "6"), "MONTH_MAP");
}
@MapExecutor(taskName = "MONTH_MAP")
public ExecuteResult monthMapExecute(MapArgs mapArgs) {
System.out.println(mapArgs);
String mapResult = mapArgs.getMapResult();
return ExecuteResult.success(mapResult);
}
@MapExecutor(taskName = "LAST_MAP")
public ExecuteResult lastMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
System.out.println(mapArgs);
return ExecuteResult.success();
return ExecuteResult.success(Integer.parseInt((String) mapArgs.getMapResult()) * 2);
}
@ReduceExecutor
public ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) {
System.out.println(mapReduceArgs);
List<String> mapResult = (List<String>) mapReduceArgs.getMapResult();
return ExecuteResult.success(mapResult.stream().map(Integer::parseInt).mapToInt(Integer::intValue).sum());
return ExecuteResult.success(
mapReduceArgs.getMapResult()
.stream()
.map(String::valueOf)
.map(Integer::parseInt)
.mapToInt(Integer::intValue).sum());
}
/**
@ -56,6 +53,11 @@ public class TestAnnoMapReduceJobExecutor {
@MergeReduceExecutor
public ExecuteResult mergeReduceExecute(MergeReduceArgs mergeReduceArgs) {
System.out.println(mergeReduceArgs);
return ExecuteResult.success();
return ExecuteResult.success(
mergeReduceArgs.getReduces()
.stream()
.map(String::valueOf)
.map(Integer::parseInt)
.mapToInt(Integer::intValue).sum());
}
}