com.google.devtools.build.lib.skyframe.RecursivePkgValueRootPackageExtractor.java Source code

Introduction

Here is the source code for com.google.devtools.build.lib.skyframe.RecursivePkgValueRootPackageExtractor.java

Source

// 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.ImmutableSet;
import com.google.common.collect.Iterables;
import com.google.devtools.build.lib.cmdline.RepositoryName;
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.LinkedHashSet;
import java.util.List;

/** Looks up {@link RecursivePkgValue}s of given roots in a {@link WalkableGraph}. */
public class RecursivePkgValueRootPackageExtractor implements RootPackageExtractor {

    public Iterable<PathFragment> getPackagesFromRoots(WalkableGraph graph, List<Root> roots,
            ExtendedEventHandler eventHandler, RepositoryName repository, PathFragment directory,
            ImmutableSet<PathFragment> blacklistedSubdirectories, ImmutableSet<PathFragment> excludedSubdirectories)
            throws InterruptedException {
        LinkedHashSet<PathFragment> packageNames = new LinkedHashSet<>();
        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), blacklistedSubdirectories));
            Preconditions.checkState(lookup != null, "Root %s in repository %s could not be found in the graph.",
                    root.asPath(), repository.getName());
            for (String packageName : lookup.getPackages()) {
                // TODO(bazel-team): Make RecursivePkgValue return NestedSet<PathFragment> so this transform
                // is unnecessary.
                PathFragment packageNamePathFragment = PathFragment.create(packageName);
                if (!Iterables.any(excludedSubdirectories,
                        excludedSubdirectory -> packageNamePathFragment.startsWith(excludedSubdirectory))) {
                    packageNames.add(packageNamePathFragment);
                }
            }
        }

        return packageNames;
    }
}