mirrors/asyncTool
1
0
Fork 0
mirror of https://gitee.com/jd-platform-opensource/asyncTool.git synced 2024-12-31 15:35:34 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

让同步执行结果有个返回值

Browse Source
This commit is contained in:
wuweifeng10 2019-12-27 16:23:39 +08:00
parent 8278337a4a
commit b7c36196f4
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/main/java/com/tianyalei/async/executor/Async.java
View File

@ -24,9 +24,9 @@ public class Async {
new LinkedBlockingQueue<>(),
(ThreadFactory) Thread::new);
public static void beginWork(long timeout, ThreadPoolExecutor pool, WorkerWrapper... workerWrapper) throws ExecutionException, InterruptedException {
public static boolean beginWork(long timeout, ThreadPoolExecutor pool, WorkerWrapper... workerWrapper) throws ExecutionException, InterruptedException {
if(workerWrapper == null || workerWrapper.length == 0) {
return;
return false;
}
List<WorkerWrapper> workerWrappers = Arrays.stream(workerWrapper).collect(Collectors.toList());
@ -37,20 +37,22 @@ public class Async {
}
try {
CompletableFuture.allOf(futures).get(timeout, TimeUnit.MILLISECONDS);
return true;
} catch (TimeoutException e) {
Set<WorkerWrapper> set = new HashSet<>();
totalWorkers(workerWrappers, set);
for (WorkerWrapper wrapper : set) {
wrapper.stopNow();
}
return false;
}
}
/**
* 同步阻塞,直到所有都完成,或失败
*/
public static void beginWork(long timeout, WorkerWrapper... workerWrapper) throws ExecutionException, InterruptedException {
beginWork(timeout, COMMON_POOL, workerWrapper);
public static boolean beginWork(long timeout, WorkerWrapper... workerWrapper) throws ExecutionException, InterruptedException {
return beginWork(timeout, COMMON_POOL, workerWrapper);
}
/**
@ -63,8 +65,12 @@ public class Async {
IGroupCallback finalGroupCallback = groupCallback;
CompletableFuture.runAsync(() -> {
try {
beginWork(timeout, COMMON_POOL, workerWrapper);
boolean success = beginWork(timeout, COMMON_POOL, workerWrapper);
if (success) {
finalGroupCallback.success(Arrays.asList(workerWrapper));
} else {
finalGroupCallback.failure(Arrays.asList(workerWrapper), new TimeoutException());
}
} catch (ExecutionException | InterruptedException e) {
e.printStackTrace();
finalGroupCallback.failure(Arrays.asList(workerWrapper), e);