gtsoft/gtsoft-snail-job-demo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化Map测试用例

Browse Source
This commit is contained in:
byteblogs168 2024-06-26 18:11:49 +08:00
parent 8664703917
commit 2e6503f403
5 changed files with 103 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/main/java/com/example/snailjob/job/TestAnnoMapJobExecutor.java Normal file
View File

@ -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();
// }
}

55
src/main/java/com/example/snailjob/job/TestAnnoMapReduceJobExecutor.java Normal file
View File

@ -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();
}
}

5
src/main/java/com/example/snailjob/job/TestMapJobExecutor.java
View File

@ -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的任务，则说明当前需要进行处理

5
src/main/java/com/example/snailjob/job/TestMapReduceJobExecutor.java
View File

@ -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的任务，则说明当前需要进行处理

2
src/main/java/com/example/snailjob/job/TestPartitionJobExecutor.java
View File

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