优化Map测试用例

This commit is contained in:
byteblogs168 2024-06-27 11:02:53 +08:00
parent 550e263d99
commit b170277f0f

View File

@ -27,27 +27,24 @@ public class TestAnnoMapReduceJobExecutor {
@MapExecutor @MapExecutor
public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) { public ExecuteResult rootMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
System.out.println(mapArgs); System.out.println(mapArgs);
return mapHandler.doMap(Lists.newArrayList(1, 2, 3, 4, 5, 6), "MONTH_MAP"); return mapHandler.doMap(Lists.newArrayList("1", "2", "3", "4", "5", "6"), "MONTH_MAP");
} }
@MapExecutor(taskName = "MONTH_MAP") @MapExecutor(taskName = "MONTH_MAP")
public ExecuteResult monthMapExecute(MapArgs mapArgs) { public ExecuteResult monthMapExecute(MapArgs mapArgs) {
System.out.println(mapArgs); System.out.println(mapArgs);
String mapResult = mapArgs.getMapResult(); return ExecuteResult.success(Integer.parseInt((String) mapArgs.getMapResult()) * 2);
return ExecuteResult.success(mapResult);
}
@MapExecutor(taskName = "LAST_MAP")
public ExecuteResult lastMapExecute(MapArgs mapArgs, MapHandler mapHandler) {
System.out.println(mapArgs);
return ExecuteResult.success();
} }
@ReduceExecutor @ReduceExecutor
public ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) { public ExecuteResult reduceExecute(ReduceArgs mapReduceArgs) {
System.out.println(mapReduceArgs); System.out.println(mapReduceArgs);
List<String> mapResult = (List<String>) mapReduceArgs.getMapResult(); return ExecuteResult.success(
return ExecuteResult.success(mapResult.stream().map(Integer::parseInt).mapToInt(Integer::intValue).sum()); mapReduceArgs.getMapResult()
.stream()
.map(String::valueOf)
.map(Integer::parseInt)
.mapToInt(Integer::intValue).sum());
} }
/** /**
@ -56,6 +53,11 @@ public class TestAnnoMapReduceJobExecutor {
@MergeReduceExecutor @MergeReduceExecutor
public ExecuteResult mergeReduceExecute(MergeReduceArgs mergeReduceArgs) { public ExecuteResult mergeReduceExecute(MergeReduceArgs mergeReduceArgs) {
System.out.println(mergeReduceArgs); System.out.println(mergeReduceArgs);
return ExecuteResult.success(); return ExecuteResult.success(
mergeReduceArgs.getReduces()
.stream()
.map(String::valueOf)
.map(Integer::parseInt)
.mapToInt(Integer::intValue).sum());
} }
} }