SettableListenableFuture properly rethrows Error

Issue: SPR-14298
This commit is contained in:
Juergen Hoeller
2016-05-26 19:39:22 +02:00
parent dacc31e3d3
commit a979885d17
2 changed files with 36 additions and 27 deletions

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2014 the original author or authors.
* Copyright 2002-2016 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -23,6 +23,7 @@ import java.util.concurrent.TimeoutException;
import java.util.concurrent.atomic.AtomicReference;
import org.springframework.util.Assert;
import org.springframework.util.ReflectionUtils;
/**
* A {@link org.springframework.util.concurrent.ListenableFuture ListenableFuture}
@@ -49,11 +50,11 @@ public class SettableListenableFuture<T> implements ListenableFuture<T> {
/**
* Set the value of this future. This method will return {@code true} if
* the value was set successfully, or {@code false} if the future has already
* been set or cancelled.
* @param value the value that will be set.
* @return {@code true} if the value was successfully set, else {@code false}.
* Set the value of this future. This method will return {@code true} if the
* value was set successfully, or {@code false} if the future has already been
* set or cancelled.
* @param value the value that will be set
* @return {@code true} if the value was successfully set, else {@code false}
*/
public boolean set(T value) {
boolean success = this.settableTask.setValue(value);
@@ -64,14 +65,14 @@ public class SettableListenableFuture<T> implements ListenableFuture<T> {
}
/**
* Set the exception of this future. This method will return {@code true} if
* the exception was set successfully, or {@code false} if the future has already
* been set or cancelled.
* @param exception the value that will be set.
* @return {@code true} if the exception was successfully set, else {@code false}.
* Set the exception of this future. This method will return {@code true} if the
* exception was set successfully, or {@code false} if the future has already been
* set or cancelled.
* @param exception the value that will be set
* @return {@code true} if the exception was successfully set, else {@code false}
*/
public boolean setException(Throwable exception) {
Assert.notNull(exception, "'exception' must not be null");
Assert.notNull(exception, "Exception must not be null");
boolean success = this.settableTask.setException(exception);
if (success) {
this.listenableFuture.run();
@@ -149,7 +150,7 @@ public class SettableListenableFuture<T> implements ListenableFuture<T> {
private static class SettableTask<T> implements Callable<T> {
private static final String NO_VALUE = SettableListenableFuture.class.getName() + ".NO_VALUE";
private static final Object NO_VALUE = new Object();
private final AtomicReference<Object> value = new AtomicReference<Object>(NO_VALUE);
@@ -176,10 +177,11 @@ public class SettableListenableFuture<T> implements ListenableFuture<T> {
@SuppressWarnings("unchecked")
@Override
public T call() throws Exception {
if (value.get() instanceof Exception) {
throw (Exception) value.get();
Object val = this.value.get();
if (val instanceof Throwable) {
ReflectionUtils.rethrowException((Throwable) val);
}
return (T) value.get();
return (T) val;
}
}