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

Introduction

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

Source

// Copyright 2016 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.MoreObjects;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.devtools.build.lib.cmdline.RepositoryName;
import com.google.devtools.build.lib.concurrent.ThreadSafety.ThreadSafe;
import com.google.devtools.build.lib.skyframe.RecursivePkgValue.RecursivePkgKey;
import com.google.devtools.build.lib.vfs.PathFragment;
import com.google.devtools.build.lib.vfs.RootedPath;
import com.google.devtools.build.skyframe.SkyKey;
import com.google.devtools.build.skyframe.SkyValue;

import java.util.Objects;

/**
 * The value computed by {@link CollectPackagesUnderDirectoryFunction}. Contains a mapping for all
 * its non-excluded directories to whether there are packages beneath them.
 *
 * <p>This value is used by {@link GraphBackedRecursivePackageProvider#getPackagesUnderDirectory}
 * to help it traverse the graph and find the set of packages under a directory, recursively by
 * {@link CollectPackagesUnderDirectoryFunction} which computes a value for a directory by
 * aggregating results calculated from its subdirectories, and by
 * {@link PrepareDepsOfTargetsUnderDirectoryFunction} which uses this value to find transitive
 * targets to load.
 *
 * <p>Note that even though the {@link CollectPackagesUnderDirectoryFunction} is evaluated in
 * part because of its side-effects (i.e. loading transitive dependencies of targets), this value
 * interacts safely with change pruning, despite the fact that this value is a lossy representation
 * of the packages beneath a directory (i.e. it doesn't care <b>which</b> packages are under a
 * directory, just whether there are any). When the targets in a package change, the
 * {@link PackageValue} that {@link CollectPackagesUnderDirectoryFunction} depends on will be
 * invalidated, and the PrepareDeps function for that package's directory will be reevaluated,
 * loading any new transitive dependencies. Change pruning may prevent the reevaluation of
 * PrepareDeps for directories above that one, but they don't need to be re-run.
 */
public class CollectPackagesUnderDirectoryValue implements SkyValue {
    public static final CollectPackagesUnderDirectoryValue EMPTY = new CollectPackagesUnderDirectoryValue(false,
            ImmutableMap.<RootedPath, Boolean>of());

    private final boolean isDirectoryPackage;
    private final ImmutableMap<RootedPath, Boolean> subdirectoryTransitivelyContainsPackages;

    private CollectPackagesUnderDirectoryValue(boolean isDirectoryPackage,
            ImmutableMap<RootedPath, Boolean> subdirectoryTransitivelyContainsPackages) {
        this.subdirectoryTransitivelyContainsPackages = subdirectoryTransitivelyContainsPackages;
        this.isDirectoryPackage = isDirectoryPackage;
    }

    public static CollectPackagesUnderDirectoryValue of(boolean isDirectoryPackage,
            ImmutableMap<RootedPath, Boolean> subdirectoryTransitivelyContainsPackages) {
        if (!isDirectoryPackage && subdirectoryTransitivelyContainsPackages.isEmpty()) {
            return EMPTY;
        }
        return new CollectPackagesUnderDirectoryValue(isDirectoryPackage, subdirectoryTransitivelyContainsPackages);
    }

    public boolean isDirectoryPackage() {
        return isDirectoryPackage;
    }

    public ImmutableMap<RootedPath, Boolean> getSubdirectoryTransitivelyContainsPackages() {
        return subdirectoryTransitivelyContainsPackages;
    }

    @Override
    public int hashCode() {
        return Objects.hash(isDirectoryPackage, subdirectoryTransitivelyContainsPackages);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (!(o instanceof CollectPackagesUnderDirectoryValue)) {
            return false;
        }
        CollectPackagesUnderDirectoryValue that = (CollectPackagesUnderDirectoryValue) o;
        return this.isDirectoryPackage == that.isDirectoryPackage && this.subdirectoryTransitivelyContainsPackages
                .equals(that.subdirectoryTransitivelyContainsPackages);
    }

    @Override
    public String toString() {
        return MoreObjects.toStringHelper(this).add("isDirectoryPackage", isDirectoryPackage)
                .add("subdirectoryTransitivelyContainsPackages", subdirectoryTransitivelyContainsPackages)
                .toString();
    }

    /** Create a collect packages under directory request. */
    @ThreadSafe
    static SkyKey key(RepositoryName repository, RootedPath rootedPath, ImmutableSet<PathFragment> excludedPaths) {
        return key(new RecursivePkgKey(repository, rootedPath, excludedPaths));
    }

    static SkyKey key(RecursivePkgKey recursivePkgKey) {
        return SkyKey.create(SkyFunctions.COLLECT_PACKAGES_UNDER_DIRECTORY, recursivePkgKey);
    }
}