diff --git a/src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java b/src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java index d013471..b275ad9 100644 --- a/src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java +++ b/src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java @@ -19,18 +19,21 @@ public class TestAnnoMapJobExecutor { @MapExecutor public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) { System.out.println(mapArgs); + System.out.println(mapArgs.getWfContext()); return mapHandler.doMap(Lists.newArrayList("1", "2", "3"), "MONTH_MAP"); } - @MapExecutor(taskName = "MONTH_MAP1") + @MapExecutor(taskName = "MONTH_MAP") public ExecuteResult monthMapExecute(MapArgs mapArgs) { System.out.println(mapArgs); + System.out.println(mapArgs.getWfContext()); return ExecuteResult.success(123); } @MapExecutor(taskName = "LAST_MAP") public ExecuteResult lastMapExecute(MapArgs mapArgs, MapHandler mapHandler) { System.out.println(mapArgs); + System.out.println(mapArgs.getWfContext()); return ExecuteResult.success(); } diff --git a/src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java b/src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java index 06c42cf..75aeb12 100644 --- a/src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java +++ b/src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java @@ -27,18 +27,23 @@ public class TestAnnoMapReduceJobExecutor { @MapExecutor public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) { 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"); } @MapExecutor(taskName = "MONTH_MAP") public ExecuteResult monthMapExecute(MapArgs mapArgs) { System.out.println(mapArgs); + System.out.println(mapArgs.getWfContext()); + mapArgs.appendContext("age", "111"); return ExecuteResult.success(Integer.parseInt((String) mapArgs.getMapResult()) * 2); } @ReduceExecutor public ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) { System.out.println(mapReduceArgs); + System.out.println(mapReduceArgs.getWfContext()); return ExecuteResult.success( mapReduceArgs.getMapResult() .stream() @@ -53,6 +58,7 @@ public class TestAnnoMapReduceJobExecutor { @MergeReduceExecutor public ExecuteResult mergeReduceExecute(MergeReduceArgs mergeReduceArgs) { System.out.println(mergeReduceArgs); + System.out.println(mergeReduceArgs.getWfContext()); return ExecuteResult.success( mergeReduceArgs.getReduces() .stream()