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-28 15:02:15 +08:00
parent d29337bd12
commit ccfdc14f74
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java
View File

@ -19,18 +19,21 @@ public class TestAnnoMapJobExecutor {
@MapExecutor @MapExecutor
public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) { public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
System.out.println(mapArgs); System.out.println(mapArgs);
System.out.println(mapArgs.getWfContext());
return mapHandler.doMap(Lists.newArrayList("1", "2", "3"), "MONTH_MAP"); return mapHandler.doMap(Lists.newArrayList("1", "2", "3"), "MONTH_MAP");
} }
@MapExecutor(taskName = "MONTH_MAP1") @MapExecutor(taskName = "MONTH_MAP")
public ExecuteResult monthMapExecute(MapArgs mapArgs) { public ExecuteResult monthMapExecute(MapArgs mapArgs) {
System.out.println(mapArgs); System.out.println(mapArgs);
System.out.println(mapArgs.getWfContext());
return ExecuteResult.success(123); return ExecuteResult.success(123);
} }
@MapExecutor(taskName = "LAST_MAP") @MapExecutor(taskName = "LAST_MAP")
public ExecuteResult lastMapExecute(MapArgs mapArgs, MapHandler mapHandler) { public ExecuteResult lastMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
System.out.println(mapArgs); System.out.println(mapArgs);
System.out.println(mapArgs.getWfContext());
return ExecuteResult.success(); return ExecuteResult.success();
} }

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

@ -27,18 +27,23 @@ public class TestAnnoMapReduceJobExecutor {
@MapExecutor @MapExecutor
public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) { public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
System.out.println(mapArgs); System.out.println(mapArgs);
System.out.println(mapArgs.getWfContext());
mapArgs.appendContext("name", "zsg");
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") @MapExecutor(taskName = "MONTH_MAP")
public ExecuteResult monthMapExecute(MapArgs mapArgs) { public ExecuteResult monthMapExecute(MapArgs mapArgs) {
System.out.println(mapArgs); System.out.println(mapArgs);
System.out.println(mapArgs.getWfContext());
mapArgs.appendContext("age", "111");
return ExecuteResult.success(Integer.parseInt((String) mapArgs.getMapResult()) * 2); return ExecuteResult.success(Integer.parseInt((String) mapArgs.getMapResult()) * 2);
} }
@ReduceExecutor @ReduceExecutor
public ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) { public ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) {
System.out.println(mapReduceArgs); System.out.println(mapReduceArgs);
System.out.println(mapReduceArgs.getWfContext());
return ExecuteResult.success( return ExecuteResult.success(
mapReduceArgs.getMapResult() mapReduceArgs.getMapResult()
.stream() .stream()
@ -53,6 +58,7 @@ public class TestAnnoMapReduceJobExecutor {
@MergeReduceExecutor @MergeReduceExecutor
public ExecuteResult mergeReduceExecute(MergeReduceArgs mergeReduceArgs) { public ExecuteResult mergeReduceExecute(MergeReduceArgs mergeReduceArgs) {
System.out.println(mergeReduceArgs); System.out.println(mergeReduceArgs);
System.out.println(mergeReduceArgs.getWfContext());
return ExecuteResult.success( return ExecuteResult.success(
mergeReduceArgs.getReduces() mergeReduceArgs.getReduces()
.stream() .stream()