优化Map测试用例

simbabdytech · Jun 28, 2024 · ccfdc14 · ccfdc14
1 parent d29337b
commit ccfdc14
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git 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
@@ -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 ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) {
     @MergeReduceExecutor
     public ExecuteResult mergeReduceExecute(MergeReduceArgs mergeReduceArgs) {
         System.out.println(mergeReduceArgs);
+        System.out.println(mergeReduceArgs.getWfContext());
         return ExecuteResult.success(
             mergeReduceArgs.getReduces()
                 .stream()