Replace Futures.dereference with the appropriate async method.
Futures.dereference is being deprecated and deleted, because it has a race condition where cancellation is not propagated.
PiperOrigin-RevId: 172748120
diff --git a/src/main/java/com/google/devtools/build/lib/query2/SkyQueryEnvironment.java b/src/main/java/com/google/devtools/build/lib/query2/SkyQueryEnvironment.java
index 1374652..e1e0344 100644
--- a/src/main/java/com/google/devtools/build/lib/query2/SkyQueryEnvironment.java
+++ b/src/main/java/com/google/devtools/build/lib/query2/SkyQueryEnvironment.java
@@ -31,6 +31,7 @@
import com.google.common.collect.Maps;
import com.google.common.collect.Multimap;
import com.google.common.collect.Sets;
+import com.google.common.util.concurrent.AsyncCallable;
import com.google.common.util.concurrent.AsyncFunction;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
@@ -580,6 +581,14 @@
}
}
+ private <R> ListenableFuture<R> safeSubmitAsync(AsyncCallable<R> callable) {
+ try {
+ return Futures.submitAsync(callable, executor);
+ } catch (RejectedExecutionException e) {
+ return Futures.immediateCancelledFuture();
+ }
+ }
+
@ThreadSafe
@Override
public QueryTaskFuture<Void> eval(
@@ -588,10 +597,9 @@
final Callback<Target> callback) {
// TODO(bazel-team): As in here, use concurrency for the async #eval of other QueryEnvironment
// implementations.
- Callable<QueryTaskFutureImpl<Void>> task =
+ AsyncCallable<Void> task =
() -> (QueryTaskFutureImpl<Void>) expr.eval(SkyQueryEnvironment.this, context, callback);
- ListenableFuture<QueryTaskFutureImpl<Void>> futureFuture = safeSubmit(task);
- return QueryTaskFutureImpl.ofDelegate(Futures.dereference(futureFuture));
+ return QueryTaskFutureImpl.ofDelegate(safeSubmitAsync(task));
}
@Override
