diff --git a/src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java b/src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java index 9c5de47..d013471 100644 --- a/src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java +++ b/src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java @@ -3,11 +3,7 @@ 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; @@ -20,22 +16,22 @@ import org.springframework.stereotype.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(); -// } + @MapExecutor + public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) { + System.out.println(mapArgs); + return mapHandler.doMap(Lists.newArrayList("1", "2", "3"), "MONTH_MAP"); + } + + @MapExecutor(taskName = "MONTH_MAP1") + 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(); + } } diff --git a/src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java b/src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java index 27a28b1..34158b7 100644 --- a/src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java +++ b/src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java @@ -9,9 +9,13 @@ 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.aizuda.snailjob.common.core.util.JsonUtil; import com.google.common.collect.Lists; import org.springframework.stereotype.Component; +import java.util.List; +import java.util.stream.Collectors; + /** * @author: opensnail * @date : 2024-06-26 @@ -23,13 +27,14 @@ public class TestAnnoMapReduceJobExecutor { @MapExecutor public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) { System.out.println(mapArgs); - return mapHandler.doMap(Lists.newArrayList("aaa"), "MONTH_MAP"); + 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); - return ExecuteResult.success(123); + String mapResult = mapArgs.getMapResult(); + return ExecuteResult.success(mapResult); } @MapExecutor(taskName = "LAST_MAP") @@ -41,7 +46,8 @@ public class TestAnnoMapReduceJobExecutor { @ReduceExecutor public ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) { System.out.println(mapReduceArgs); - return ExecuteResult.success(); + List mapResult = (List) mapReduceArgs.getMapResult(); + return ExecuteResult.success(mapResult.stream().map(Integer::parseInt).mapToInt(Integer::intValue).sum()); } /**