commit 2643d4b7543403eae52c038e769231f539938195
author shreyax <shreyax@google.com> Fri May 25 11:12:11 2018 -0700
committer Copybara-Service <copybara-piper@google.com> Fri May 25 11:13:54 2018 -0700
tree 9a6ddcec6b099af4187faa0e86c127e0001ba4d0
parent af18c36dc46bfe2334fca442f6d8cc9f6943fb7c

Implement unbounded deps() using ParallelVisitor for SkyQuery.

PiperOrigin-RevId: 198074986

src/main/java/com/google/devtools/build/lib/query2/ParallelSkyQueryUtils.java

diff --git a/src/main/java/com/google/devtools/build/lib/query2/ParallelSkyQueryUtils.java b/src/main/java/com/google/devtools/build/lib/query2/ParallelSkyQueryUtils.java
index 516d2e6..1fe3bbf 100644
--- a/src/main/java/com/google/devtools/build/lib/query2/ParallelSkyQueryUtils.java
+++ b/src/main/java/com/google/devtools/build/lib/query2/ParallelSkyQueryUtils.java
@@ -170,6 +170,22 @@
     visitor.visitAndWaitForCompletion(env.getFileStateKeysForFileFragments(fileIdentifiers));
   }
 
+  static QueryTaskFuture<Void> getDepsUnboundedParallel(
+      SkyQueryEnvironment env,
+      QueryExpression expression,
+      VariableContext<Target> context,
+      Callback<Target> callback,
+      MultisetSemaphore<PackageIdentifier> packageSemaphore,
+      boolean depsNeedFiltering,
+      Callback<Target> errorReporter) {
+    return env.eval(
+        expression,
+        context,
+        ParallelVisitor.createParallelVisitorCallback(
+            new DepsUnboundedVisitor.Factory(
+                env, callback, packageSemaphore, depsNeedFiltering, errorReporter)));
+  }
+
   static class DepAndRdep {
     @Nullable final SkyKey dep;
     final SkyKey rdep;