| // Copyright 2018 The Bazel Authors. All rights reserved. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| package com.google.devtools.build.lib.skyframe; |
| |
| import com.google.common.base.Preconditions; |
| import com.google.common.collect.ImmutableList; |
| import com.google.common.collect.ImmutableSet; |
| import com.google.common.collect.Iterables; |
| import com.google.devtools.build.lib.cmdline.PackageIdentifier; |
| import com.google.devtools.build.lib.cmdline.RepositoryName; |
| import com.google.devtools.build.lib.concurrent.ParallelVisitor.UnusedException; |
| import com.google.devtools.build.lib.concurrent.ThreadSafeBatchCallback; |
| import com.google.devtools.build.lib.events.ExtendedEventHandler; |
| import com.google.devtools.build.lib.vfs.PathFragment; |
| import com.google.devtools.build.lib.vfs.Root; |
| import com.google.devtools.build.lib.vfs.RootedPath; |
| import com.google.devtools.build.skyframe.WalkableGraph; |
| import java.util.List; |
| |
| /** Looks up {@link RecursivePkgValue}s of given roots in a {@link WalkableGraph}. */ |
| public class RecursivePkgValueRootPackageExtractor implements RootPackageExtractor { |
| |
| @Override |
| public void streamPackagesFromRoots( |
| ThreadSafeBatchCallback<PackageIdentifier, UnusedException> results, |
| WalkableGraph graph, |
| List<Root> roots, |
| ExtendedEventHandler eventHandler, |
| RepositoryName repository, |
| PathFragment directory, |
| ImmutableSet<PathFragment> blacklistedSubdirectories, |
| ImmutableSet<PathFragment> excludedSubdirectories) |
| throws InterruptedException { |
| ImmutableSet<PathFragment> filteredBlacklistedSubdirectories = |
| ImmutableSet.copyOf( |
| Iterables.filter( |
| blacklistedSubdirectories, |
| path -> !path.equals(directory) && path.startsWith(directory))); |
| |
| for (Root root : roots) { |
| // Note: no need to check if lookup == null because it will never be null. |
| // {@link RecursivePkgFunction} handles all errors in a keep_going build. |
| // In a nokeep_going build, we would never reach this part of the code. |
| RecursivePkgValue lookup = |
| (RecursivePkgValue) |
| graph.getValue( |
| RecursivePkgValue.key( |
| repository, |
| RootedPath.toRootedPath(root, directory), |
| filteredBlacklistedSubdirectories)); |
| Preconditions.checkState( |
| lookup != null, |
| "Root %s in repository %s could not be found in the graph.", |
| root.asPath(), |
| repository.getName()); |
| ImmutableList.Builder<PackageIdentifier> packageIds = ImmutableList.builder(); |
| for (String packageName : lookup.getPackages().toList()) { |
| // TODO(bazel-team): Make RecursivePkgValue return NestedSet<PathFragment> so this transform |
| // is unnecessary. |
| PathFragment packageNamePathFragment = PathFragment.create(packageName); |
| if (!Iterables.any(excludedSubdirectories, packageNamePathFragment::startsWith)) { |
| packageIds.add(PackageIdentifier.create(repository, packageNamePathFragment)); |
| } |
| } |
| results.process(packageIds.build()); |
| } |
| } |
| } |