From ccfdc14f74b6eb343e6df71e561d2a541634658a Mon Sep 17 00:00:00 2001
From: byteblogs168 <598092184@qq.com>
Date: Fri, 28 Jun 2024 15:02:15 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96Map=E6=B5=8B=E8=AF=95?=
 =?UTF-8?q?=E7=94=A8=E4=BE=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/example/snailjob/job/TestAnnoMapJobExecutor.java    | 5 ++++-
 .../example/snailjob/job/TestAnnoMapReduceJobExecutor.java  | 6 ++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

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()