fix ThreadUtils.await() blocking on exception
This commit is contained in:
parent
847e9e8701
commit
11c8b05f45
@ -48,14 +48,25 @@ public class ThreadUtils {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Awaits execution of the given command, but does not throw its exception.
|
||||||
|
*
|
||||||
|
* @param command the command to execute
|
||||||
|
* @param threadId the thread id
|
||||||
|
*/
|
||||||
public static void await(Runnable command, String threadId) {
|
public static void await(Runnable command, String threadId) {
|
||||||
if (isCurrentThread(Thread.currentThread(), threadId)) {
|
if (isCurrentThread(Thread.currentThread(), threadId)) {
|
||||||
command.run();
|
command.run();
|
||||||
} else {
|
} else {
|
||||||
CountDownLatch latch = new CountDownLatch(1);
|
CountDownLatch latch = new CountDownLatch(1);
|
||||||
execute(() -> {
|
execute(() -> {
|
||||||
command.run();
|
try {
|
||||||
latch.countDown();
|
command.run();
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw e;
|
||||||
|
} finally {
|
||||||
|
latch.countDown();
|
||||||
|
}
|
||||||
}, threadId);
|
}, threadId);
|
||||||
try {
|
try {
|
||||||
latch.await();
|
latch.await();
|
||||||
|
Loading…
Reference in New Issue
Block a user