84 lines
2.2 KiB
Java
84 lines
2.2 KiB
Java
package mark.quinn.async;
|
|
|
|
import java.util.concurrent.Callable;
|
|
import java.util.concurrent.Future;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.beans.factory.DisposableBean;
|
|
import org.springframework.beans.factory.InitializingBean;
|
|
import org.springframework.core.task.AsyncTaskExecutor;
|
|
|
|
public class ExceptionHandlingAsyncTaskExecutor implements AsyncTaskExecutor,
|
|
InitializingBean, DisposableBean {
|
|
|
|
private final Logger log = LoggerFactory.getLogger(ExceptionHandlingAsyncTaskExecutor.class);
|
|
|
|
private final AsyncTaskExecutor executor;
|
|
|
|
public ExceptionHandlingAsyncTaskExecutor(AsyncTaskExecutor executor) {
|
|
this.executor = executor;
|
|
}
|
|
|
|
@Override
|
|
public void execute(Runnable task) {
|
|
executor.execute(task);
|
|
}
|
|
|
|
@Override
|
|
public void execute(Runnable task, long startTimeout) {
|
|
executor.execute(createWrappedRunnable(task), startTimeout);
|
|
}
|
|
|
|
private <T> Callable<T> createCallable(final Callable<T> task) {
|
|
return () -> {
|
|
try {
|
|
return task.call();
|
|
} catch (Exception e) {
|
|
handle(e);
|
|
throw e;
|
|
}
|
|
};
|
|
}
|
|
|
|
private Runnable createWrappedRunnable(final Runnable task) {
|
|
return () -> {
|
|
try {
|
|
task.run();
|
|
} catch (Exception e) {
|
|
handle(e);
|
|
}
|
|
};
|
|
}
|
|
|
|
protected void handle(Exception e) {
|
|
log.error("Caught async exception", e);
|
|
}
|
|
|
|
@Override
|
|
public Future<?> submit(Runnable task) {
|
|
return executor.submit(createWrappedRunnable(task));
|
|
}
|
|
|
|
@Override
|
|
public <T> Future<T> submit(Callable<T> task) {
|
|
return executor.submit(createCallable(task));
|
|
}
|
|
|
|
@Override
|
|
public void destroy() throws Exception {
|
|
if (executor instanceof DisposableBean) {
|
|
DisposableBean bean = (DisposableBean) executor;
|
|
bean.destroy();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void afterPropertiesSet() throws Exception {
|
|
if (executor instanceof InitializingBean) {
|
|
InitializingBean bean = (InitializingBean) executor;
|
|
bean.afterPropertiesSet();
|
|
}
|
|
}
|
|
}
|