Expose build file label retrieval as a protected member.

PiperOrigin-RevId: 191752718

1 files changed, 12 insertions, 13 deletions

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 c96a9ecc71..0c8d656c10 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
@@ -760,19 +760,7 @@ public class SkyQueryEnvironment extends AbstractBlazeQueryEnvironment<Target>
 
           // Also add the BUILD file of the extension.
           if (buildFiles) {
-            Path buildFileForLoad = null;
-            try {
-              buildFileForLoad =
-                  pkgPath.getPackageBuildFile(loadTarget.getLabel().getPackageIdentifier());
-            } catch (NoSuchPackageException e) {
-              throw new QueryException(
-                  loadTarget.getLabel().getPackageIdentifier() + " does not exist in graph");
-            }
-            Label buildFileLabel =
-                Label.createUnvalidated(
-                    loadTarget.getLabel().getPackageIdentifier(),
-                    buildFileForLoad.getBaseName());
-
+            Label buildFileLabel = getBuildFileLabel(loadTarget.getLabel().getPackageIdentifier());
             addIfUniqueLabel(new FakeLoadTarget(buildFileLabel, pkg), seenLabels, dependentFiles);
           }
         }
@@ -781,6 +769,17 @@ public class SkyQueryEnvironment extends AbstractBlazeQueryEnvironment<Target>
     return dependentFiles;
   }
 
+  protected Label getBuildFileLabel(PackageIdentifier packageIdentifier) throws QueryException {
+    // TODO(bazel-team): Try avoid filesystem access here.
+    Path buildFileForLoad = null;
+    try {
+      buildFileForLoad = pkgPath.getPackageBuildFile(packageIdentifier);
+    } catch (NoSuchPackageException e) {
+      throw new QueryException(packageIdentifier + " does not exist in graph");
+    }
+    return Label.createUnvalidated(packageIdentifier, buildFileForLoad.getBaseName());
+  }
+
   private static void addIfUniqueLabel(Target node, Set<Label> labels, Set<Target> nodes) {
     if (labels.add(node.getLabel())) {
       nodes.add(node);