优化Map测试用例

src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java

@@ -0,0 +1,41 @@
+package com.example.snailjob.job;
+
+import com.aizuda.snailjob.client.job.core.MapHandler;
+import com.aizuda.snailjob.client.job.core.annotation.JobExecutor;
+import com.aizuda.snailjob.client.job.core.annotation.MapExecutor;
+import com.aizuda.snailjob.client.job.core.annotation.MergeReduceExecutor;
+import com.aizuda.snailjob.client.job.core.annotation.ReduceExecutor;
+import com.aizuda.snailjob.client.job.core.dto.MapArgs;
+import com.aizuda.snailjob.client.job.core.dto.MergeReduceArgs;
+import com.aizuda.snailjob.client.job.core.dto.ReduceArgs;
+import com.aizuda.snailjob.client.model.ExecuteResult;
+import com.google.common.collect.Lists;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author: opensnail
+ * @date : 2024-06-26
+ */
+@Component
+@JobExecutor(name = "testAnnoMapJobExecutor")
+public class TestAnnoMapJobExecutor {
+
+//    @MapExecutor
+//    public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
+//        System.out.println(mapArgs);
+//        return mapHandler.doMap(Lists.newArrayList("aaa"), "MONTH_MAP");
+//    }
+//
+//    @MapExecutor(taskName = "MONTH_MAP")
+//    public ExecuteResult monthMapExecute(MapArgs mapArgs) {
+//        System.out.println(mapArgs);
+//        return ExecuteResult.success(123);
+//    }
+//
+//    @MapExecutor(taskName = "LAST_MAP")
+//    public ExecuteResult lastMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
+//        System.out.println(mapArgs);
+//        return ExecuteResult.success();
+//    }
+
+}

src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java

@@ -0,0 +1,55 @@
+package com.example.snailjob.job;
+
+import com.aizuda.snailjob.client.job.core.MapHandler;
+import com.aizuda.snailjob.client.job.core.annotation.JobExecutor;
+import com.aizuda.snailjob.client.job.core.annotation.MapExecutor;
+import com.aizuda.snailjob.client.job.core.annotation.MergeReduceExecutor;
+import com.aizuda.snailjob.client.job.core.annotation.ReduceExecutor;
+import com.aizuda.snailjob.client.job.core.dto.MapArgs;
+import com.aizuda.snailjob.client.job.core.dto.MergeReduceArgs;
+import com.aizuda.snailjob.client.job.core.dto.ReduceArgs;
+import com.aizuda.snailjob.client.model.ExecuteResult;
+import com.google.common.collect.Lists;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author: opensnail
+ * @date : 2024-06-26
+ */
+@Component
+@JobExecutor(name = "testAnnoMapReduceJobExecutor")
+public class TestAnnoMapReduceJobExecutor {
+
+    @MapExecutor
+    public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
+        System.out.println(mapArgs);
+        return mapHandler.doMap(Lists.newArrayList("aaa"), "MONTH_MAP");
+    }
+
+    @MapExecutor(taskName = "MONTH_MAP")
+    public ExecuteResult monthMapExecute(MapArgs mapArgs) {
+        System.out.println(mapArgs);
+        return ExecuteResult.success(123);
+    }
+
+    @MapExecutor(taskName = "LAST_MAP")
+    public ExecuteResult lastMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
+        System.out.println(mapArgs);
+        return ExecuteResult.success();
+    }
+
+    @ReduceExecutor
+    public ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) {
+        System.out.println(mapReduceArgs);
+        return ExecuteResult.success();
+    }
+
+    /**
+     * 当只有一个reduce任务时无此执行器
+     */
+    @MergeReduceExecutor
+    public ExecuteResult mergeReduceExecute(MergeReduceArgs mergeReduceArgs) {
+        System.out.println(mergeReduceArgs);
+        return ExecuteResult.success();
+    }
+}

src/main/java/com/example/snailjob/job/TestMapJobExecutor.java

@@ -1,5 +1,6 @@
 package com.example.snailjob.job;
 
+import com.aizuda.snailjob.client.job.core.MapHandler;
 import com.aizuda.snailjob.client.job.core.dto.MapArgs;
 
 import com.aizuda.snailjob.client.job.core.executor.AbstractMapExecutor;
@@ -30,7 +31,7 @@ import java.util.Random;
 public class TestMapJobExecutor extends AbstractMapExecutor {
 
     @Override
-    public ExecuteResult doJobMapExecute(MapArgs mapArgs) {
+    public ExecuteResult doJobMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
         MapEnum mapEnum = MapEnum.ofMap(mapArgs.getTaskName());
         if (Objects.nonNull(mapEnum) && Objects.nonNull(mapEnum.getMap())) {
            Map map = mapEnum.getMap();
@@ -40,7 +41,7 @@ public class TestMapJobExecutor extends AbstractMapExecutor {
                 nextName = nextMap.name();
             }
 
-            return doMap(map.map(mapArgs), nextName);
+            return mapHandler.doMap(map.map(mapArgs), nextName);
         }
 
         // 未找到map的任务，则说明当前需要进行处理

src/main/java/com/example/snailjob/job/TestMapReduceJobExecutor.java

@@ -1,5 +1,6 @@
 package com.example.snailjob.job;
 
+import com.aizuda.snailjob.client.job.core.MapHandler;
 import com.aizuda.snailjob.client.job.core.dto.MapArgs;
 import com.aizuda.snailjob.client.job.core.dto.MergeReduceArgs;
 import com.aizuda.snailjob.client.job.core.dto.ReduceArgs;
@@ -32,7 +33,7 @@ import java.util.Random;
 public class TestMapReduceJobExecutor extends AbstractMapReduceExecutor {
 
     @Override
-    public ExecuteResult doJobMapExecute(MapArgs mapArgs) {
+    public ExecuteResult doJobMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
         MapEnum mapEnum = MapEnum.ofMap(mapArgs.getTaskName());
         if (Objects.nonNull(mapEnum) && Objects.nonNull(mapEnum.getMap())) {
             Map map = mapEnum.getMap();
@@ -42,7 +43,7 @@ public class TestMapReduceJobExecutor extends AbstractMapReduceExecutor {
                 nextName = nextMap.name();
             }
 
-            return doMap(map.map(mapArgs), nextName);
+            return mapHandler.doMap(map.map(mapArgs), nextName);
         }
 
         // 未找到map的任务，则说明当前需要进行处理

src/main/java/com/example/snailjob/job/TestPartitionJobExecutor.java

@@ -17,7 +17,7 @@ import org.springframework.stereotype.Component;
 public class TestPartitionJobExecutor {
 
     public ExecuteResult jobExecute(JobArgs jobArgs) {
-        if (jobArgs.getArgsStr().equals("1")) {
+        if (jobArgs.getJobParams().equals("1")) {
             throw new NullPointerException("分区空指针抛异常了");
         }
         FailOrderPo failOrderPo = new FailOrderPo();