| // Copyright 2014 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.actions; | |
| import static com.google.common.collect.ImmutableList.toImmutableList; | |
| import com.google.common.annotations.VisibleForTesting; | |
| import com.google.common.base.Function; | |
| import com.google.common.base.Functions; | |
| import com.google.common.base.Joiner; | |
| import com.google.common.base.Preconditions; | |
| import com.google.common.base.Predicate; | |
| import com.google.common.collect.ImmutableList; | |
| import com.google.common.collect.Interner; | |
| import com.google.common.collect.Iterables; | |
| import com.google.common.collect.Streams; | |
| import com.google.devtools.build.lib.actions.ActionAnalysisMetadata.MiddlemanType; | |
| import com.google.devtools.build.lib.actions.ActionLookupValue.ActionLookupKey; | |
| import com.google.devtools.build.lib.actions.ArtifactResolver.ArtifactResolverSupplier; | |
| import com.google.devtools.build.lib.cmdline.Label; | |
| import com.google.devtools.build.lib.cmdline.LabelConstants; | |
| import com.google.devtools.build.lib.collect.nestedset.NestedSet; | |
| import com.google.devtools.build.lib.concurrent.BlazeInterners; | |
| import com.google.devtools.build.lib.concurrent.ThreadSafety; | |
| import com.google.devtools.build.lib.concurrent.ThreadSafety.Immutable; | |
| import com.google.devtools.build.lib.skyframe.serialization.DeserializationContext; | |
| import com.google.devtools.build.lib.skyframe.serialization.ObjectCodec; | |
| import com.google.devtools.build.lib.skyframe.serialization.SerializationContext; | |
| import com.google.devtools.build.lib.skyframe.serialization.SerializationException; | |
| import com.google.devtools.build.lib.skyframe.serialization.autocodec.AutoCodec; | |
| import com.google.devtools.build.lib.skylarkbuildapi.FileApi; | |
| import com.google.devtools.build.lib.syntax.Printer; | |
| import com.google.devtools.build.lib.syntax.SkylarkType; | |
| import com.google.devtools.build.lib.util.FileType; | |
| import com.google.devtools.build.lib.vfs.Path; | |
| import com.google.devtools.build.lib.vfs.PathFragment; | |
| import com.google.devtools.build.skyframe.ShareabilityOfValue; | |
| import com.google.devtools.build.skyframe.SkyFunctionName; | |
| import com.google.devtools.build.skyframe.SkyKey; | |
| import com.google.protobuf.CodedInputStream; | |
| import com.google.protobuf.CodedOutputStream; | |
| import java.io.IOException; | |
| import java.util.ArrayList; | |
| import java.util.Collection; | |
| import java.util.Comparator; | |
| import java.util.List; | |
| import java.util.Map; | |
| import javax.annotation.Nullable; | |
| /** | |
| * An Artifact represents a file used by the build system, whether it's a source file or a derived | |
| * (output) file. Not all Artifacts have a corresponding FileTarget object in the <code> | |
| * build.lib.packages</code> API: for example, low-level intermediaries internal to a given rule, | |
| * such as a Java class files or C++ object files. However all FileTargets have a corresponding | |
| * Artifact. | |
| * | |
| * <p>In any given call to SkyframeExecutor#buildArtifacts(), no two Artifacts in the action graph | |
| * may refer to the same path. | |
| * | |
| * <p>Artifacts generally fall into two classifications, source and derived, but there exist a few | |
| * other cases that are fuzzy and difficult to classify. The following cases exist: | |
| * | |
| * <ul> | |
| * <li>Well-formed source Artifacts will have null generating Actions and a root that is | |
| * orthogonal to execRoot. (With the root coming from the package path.) | |
| * <li>Well-formed derived Artifacts will have non-null generating Actions, and a root that is | |
| * below execRoot. | |
| * <li>Symlinked include source Artifacts under the output/include tree will appear to be derived | |
| * artifacts with null generating Actions. | |
| * <li>Some derived Artifacts, mostly in the genfiles tree and mostly discovered during include | |
| * validation, will also have null generating Actions. | |
| * </ul> | |
| * | |
| * In the usual case, an Artifact represents a single file. However, an Artifact may also represent | |
| * the following: | |
| * | |
| * <ul> | |
| * <li>A TreeArtifact, which is a directory containing a tree of unknown {@link Artifact}s. In the | |
| * future, Actions will be able to examine these files as inputs and declare them as outputs | |
| * at execution time, but this is not yet implemented. This is used for Actions where the | |
| * inputs and/or outputs might not be discoverable except during Action execution. | |
| * <li>A directory of unknown contents, but not a TreeArtifact. This is a legacy facility and | |
| * should not be used by any new rule implementations. In particular, the file system cache | |
| * integrity checks fail for directories. | |
| * <li>An 'aggregating middleman' special Artifact, which may be expanded using a {@link | |
| * ArtifactExpander} at Action execution time. This is used by a handful of rules to save | |
| * memory. | |
| * <li>A 'constant metadata' special Artifact. These represent real files, changes to which are | |
| * ignored by the build system. They are useful for files which change frequently but do not | |
| * affect the result of a build, such as timestamp files. | |
| * <li>A 'Fileset' special Artifact. This is a legacy type of Artifact and should not be used by | |
| * new rule implementations. | |
| * </ul> | |
| * | |
| * <p>While Artifact implements {@link SkyKey} for memory-saving purposes, Skyframe requests | |
| * involving artifacts should always go through {@link Artifact#key} since ordinary derived | |
| * artifacts should not be requested directly from Skyframe. | |
| */ | |
| @Immutable | |
| public abstract class Artifact | |
| implements FileType.HasFileType, | |
| ActionInput, | |
| FileApi, | |
| Comparable<Artifact>, | |
| CommandLineItem, | |
| SkyKey { | |
| public static final SkylarkType TYPE = SkylarkType.of(Artifact.class); | |
| /** Compares artifact according to their exec paths. Sorts null values first. */ | |
| @SuppressWarnings("ReferenceEquality") // "a == b" is an optimization | |
| public static final Comparator<Artifact> EXEC_PATH_COMPARATOR = | |
| (a, b) -> { | |
| if (a == b) { | |
| return 0; | |
| } else if (a == null) { | |
| return -1; | |
| } else if (b == null) { | |
| return 1; | |
| } else { | |
| return a.execPath.compareTo(b.execPath); | |
| } | |
| }; | |
| /** Compares artifact according to their root relative paths. Sorts null values first. */ | |
| @SuppressWarnings("ReferenceEquality") // "a == b" is an optimization | |
| public static final Comparator<Artifact> ROOT_RELATIVE_PATH_COMPARATOR = | |
| (a, b) -> { | |
| if (a == b) { | |
| return 0; | |
| } else if (a == null) { | |
| return -1; | |
| } else if (b == null) { | |
| return 1; | |
| } else { | |
| int result = a.getRootRelativePath().compareTo(b.getRootRelativePath()); | |
| if (result == 0) { | |
| // Use the full exec path as a fallback if the root-relative paths are the same, thus | |
| // avoiding problems when ImmutableSortedMaps are switched from EXEC_PATH_COMPARATOR. | |
| return a.execPath.compareTo(b.execPath); | |
| } else { | |
| return result; | |
| } | |
| } | |
| }; | |
| /** | |
| * {@link com.google.devtools.build.lib.skyframe.ArtifactFunction} does direct filesystem access | |
| * without declaring Skyframe dependencies if the artifact is a source directory. However, that | |
| * filesystem access is not invalidated on incremental builds, and we have no plans to fix it, | |
| * since general consumption of source directories in this way is unsound. Therefore no new bugs | |
| * are created by declaring {@link com.google.devtools.build.lib.skyframe.ArtifactFunction} to be | |
| * hermetic. | |
| * | |
| * <p>TODO(janakr): Avoid this issue entirely by giving {@link SourceArtifact} its own {@code | |
| * SkyFunction}. Then we can just declare that function to be non-hermetic. That will also save | |
| * memory since we can make mandatory source artifacts their own SkyKeys! | |
| */ | |
| public static final SkyFunctionName ARTIFACT = SkyFunctionName.createHermetic("ARTIFACT"); | |
| /** | |
| * Returns a {@link SkyKey} that, when built, will produce this artifact. For source artifacts and | |
| * generated artifacts that may aggregate other artifacts (middleman, since they may be | |
| * aggregating middlemen, and tree), returns the artifact itself. For normal generated artifacts, | |
| * returns the key of the generating action. | |
| * | |
| * <p>Callers should use this method (or the related ones below) in preference to directly | |
| * requesting an {@link Artifact} to be built by Skyframe, since ordinary derived artifacts should | |
| * never be directly built by Skyframe. | |
| */ | |
| @ThreadSafety.ThreadSafe | |
| public static SkyKey key(Artifact artifact) { | |
| if (artifact.isTreeArtifact() | |
| || artifact.isMiddlemanArtifact() | |
| || artifact.isSourceArtifact()) { | |
| return artifact; | |
| } | |
| return ((DerivedArtifact) artifact).getGeneratingActionKey(); | |
| } | |
| public static Iterable<SkyKey> keys(Iterable<Artifact> artifacts) { | |
| return Iterables.transform(artifacts, Artifact::key); | |
| } | |
| @Override | |
| public int compareTo(Artifact o) { | |
| return EXEC_PATH_COMPARATOR.compare(this, o); | |
| } | |
| /** An object that can expand middleman and tree artifacts. */ | |
| public interface ArtifactExpander { | |
| /** | |
| * Expands the given artifact, and populates "output" with the result. | |
| * | |
| * <p>{@code artifact.isMiddlemanArtifact() || artifact.isTreeArtifact()} must be true. | |
| * Only aggregating middlemen and tree artifacts are expanded. | |
| */ | |
| void expand(Artifact artifact, Collection<? super Artifact> output); | |
| /** | |
| * Retrieve the expansion of Filesets for the given artifact. | |
| * | |
| * @param artifact {@code artifact.isFileset()} must be true. | |
| */ | |
| default ImmutableList<FilesetOutputSymlink> getFileset(Artifact artifact) { | |
| throw new UnsupportedOperationException(); | |
| } | |
| } | |
| /** Implementation of {@link ArtifactExpander} */ | |
| public static class ArtifactExpanderImpl implements ArtifactExpander { | |
| private final Map<Artifact, Collection<Artifact>> expandedInputs; | |
| private final Map<Artifact, ImmutableList<FilesetOutputSymlink>> expandedFilesets; | |
| public ArtifactExpanderImpl( | |
| Map<Artifact, Collection<Artifact>> expandedInputs, | |
| Map<Artifact, ImmutableList<FilesetOutputSymlink>> expandedFilesets) { | |
| this.expandedInputs = expandedInputs; | |
| this.expandedFilesets = expandedFilesets; | |
| } | |
| @Override | |
| public void expand(Artifact artifact, Collection<? super Artifact> output) { | |
| Preconditions.checkState( | |
| artifact.isMiddlemanArtifact() || artifact.isTreeArtifact(), artifact); | |
| Collection<Artifact> result = expandedInputs.get(artifact); | |
| if (result != null) { | |
| output.addAll(result); | |
| } | |
| } | |
| @Override | |
| public ImmutableList<FilesetOutputSymlink> getFileset(Artifact artifact) { | |
| Preconditions.checkState(artifact.isFileset()); | |
| return Preconditions.checkNotNull(expandedFilesets.get(artifact)); | |
| } | |
| } | |
| public static final ImmutableList<Artifact> NO_ARTIFACTS = ImmutableList.of(); | |
| /** A Predicate that evaluates to true if the Artifact is not a middleman artifact. */ | |
| public static final Predicate<Artifact> MIDDLEMAN_FILTER = input -> !input.isMiddlemanArtifact(); | |
| private final int hashCode; | |
| private final ArtifactRoot root; | |
| private final PathFragment execPath; | |
| /** | |
| * Content-based output paths are experimental. Only derived artifacts that are explicitly opted | |
| * in by their creating rules should use them and only when {@link | |
| * com.google.devtools.build.lib.analysis.config.BuildConfiguration#useContentBasedOutputPaths} is | |
| * on. | |
| */ | |
| private final boolean contentBasedPath; | |
| private Artifact(ArtifactRoot root, PathFragment execPath, boolean contentBasedPath) { | |
| Preconditions.checkNotNull(root); | |
| // The ArtifactOwner is not part of this computation because it is very rare that two Artifacts | |
| // have the same execPath and different owners, so a collision is fine there. If this is | |
| // changed, OwnerlessArtifactWrapper must also be changed. | |
| this.hashCode = execPath.hashCode(); | |
| this.root = root; | |
| this.execPath = execPath; | |
| this.contentBasedPath = contentBasedPath; | |
| } | |
| /** An artifact corresponding to a file in the output tree, generated by an {@link Action}. */ | |
| @AutoCodec | |
| public static class DerivedArtifact extends Artifact { | |
| /** Only used for deserializing artifacts. */ | |
| private static final Interner<DerivedArtifact> INTERNER = BlazeInterners.newWeakInterner(); | |
| /** | |
| * An {@link ActionLookupKey} until {@link #setGeneratingActionKey} is set, at which point it is | |
| * an {@link ActionLookupData}, whose {@link ActionLookupData#getActionLookupKey} will be the | |
| * same as the original value of owner. | |
| * | |
| * <p>We overload this field in order to save memory. | |
| */ | |
| private Object owner; | |
| /** Standard constructor for derived artifacts. */ | |
| public DerivedArtifact(ArtifactRoot root, PathFragment execPath, ActionLookupKey owner) { | |
| this(root, execPath, owner, /*contentBasedPath=*/ false); | |
| } | |
| /** | |
| * Same as {@link #DerivedArtifact(ArtifactRoot, PathFragment, ActionLookupKey)} but includes | |
| * tge option to use a content-based path for this artifact (see {@link | |
| * com.google.devtools.build.lib.analysis.config.BuildConfiguration#useContentBasedOutputPaths}). | |
| */ | |
| public DerivedArtifact( | |
| ArtifactRoot root, PathFragment execPath, ActionLookupKey owner, boolean contentBasedPath) { | |
| super(root, execPath, contentBasedPath); | |
| Preconditions.checkState( | |
| !root.getExecPath().isEmpty(), "Derived root has no exec path: %s, %s", root, execPath); | |
| this.owner = owner; | |
| } | |
| /** | |
| * Called when a configured target's actions are being collected. {@code generatingActionKey} | |
| * must have the same owner as this artifact's current {@link #getArtifactOwner}. | |
| */ | |
| @VisibleForTesting | |
| public void setGeneratingActionKey(ActionLookupData generatingActionKey) { | |
| Preconditions.checkState( | |
| this.owner instanceof ArtifactOwner, | |
| "Already set generating action key: %s (%s %s)", | |
| this, | |
| this.owner, | |
| generatingActionKey); | |
| Preconditions.checkState( | |
| Preconditions.checkNotNull(generatingActionKey, this).getActionLookupKey().equals(owner), | |
| "Owner of generating action key not same as artifact's owner: %s (%s %s)", | |
| this, | |
| this.owner, | |
| generatingActionKey); | |
| this.owner = Preconditions.checkNotNull(generatingActionKey, this); | |
| } | |
| @VisibleForTesting | |
| public boolean hasGeneratingActionKey() { | |
| return this.owner instanceof ActionLookupData; | |
| } | |
| /** Can only be called once {@link #setGeneratingActionKey} is called. */ | |
| public ActionLookupData getGeneratingActionKey() { | |
| Preconditions.checkState(owner instanceof ActionLookupData, "Bad owner: %s %s", this, owner); | |
| return (ActionLookupData) owner; | |
| } | |
| @Override | |
| public ActionLookupValue.ActionLookupKey getArtifactOwner() { | |
| return owner instanceof ActionLookupData | |
| ? getGeneratingActionKey().getActionLookupKey() | |
| : (ActionLookupKey) owner; | |
| } | |
| @Override | |
| public Label getOwnerLabel() { | |
| return getArtifactOwner().getLabel(); | |
| } | |
| @Override | |
| public PathFragment getRootRelativePath() { | |
| return getExecPath().relativeTo(getRoot().getExecPath()); | |
| } | |
| @Override | |
| boolean ownersEqual(Artifact other) { | |
| DerivedArtifact that = (DerivedArtifact) other; | |
| if (!(this.owner instanceof ActionLookupData) || !(that.owner instanceof ActionLookupData)) { | |
| // Happens when at least one of these artifacts hasn't had its generating action key set | |
| // yet, so its configured target is still being analyzed. Tolerate. | |
| return this.getArtifactOwner().equals(that.getArtifactOwner()); | |
| } | |
| return this.owner.equals(that.owner); | |
| } | |
| /** | |
| * The {@code rootRelativePath is a few characters shorter than the {@code execPath}, so we save | |
| * a few bytes by serializing it rather than the {@code execPath}, especially when the {@code | |
| * root} is common to many artifacts and therefore memoized. | |
| */ | |
| @AutoCodec.VisibleForSerialization | |
| @AutoCodec.Instantiator | |
| static DerivedArtifact createForSerialization( | |
| ArtifactRoot root, PathFragment rootRelativePath, ActionLookupData generatingActionKey) { | |
| if (rootRelativePath == null | |
| || rootRelativePath.isAbsolute() != root.getRoot().isAbsolute()) { | |
| throw new IllegalArgumentException( | |
| rootRelativePath | |
| + ": illegal rootRelativePath for " | |
| + root | |
| + " (generatingActionKey: " | |
| + generatingActionKey | |
| + ")"); | |
| } | |
| Preconditions.checkState( | |
| !root.isSourceRoot(), "Root not derived: %s %s", root, rootRelativePath); | |
| PathFragment rootExecPath = root.getExecPath(); | |
| DerivedArtifact artifact = | |
| new DerivedArtifact( | |
| root, | |
| rootExecPath.getRelative(rootRelativePath), | |
| generatingActionKey.getActionLookupKey(), | |
| /*contentBasedPath=*/ false); | |
| artifact.setGeneratingActionKey(generatingActionKey); | |
| return INTERNER.intern(artifact); | |
| } | |
| } | |
| public final Path getPath() { | |
| return root.getRoot().getRelative(getRootRelativePath()); | |
| } | |
| public boolean hasParent() { | |
| return getParent() != null; | |
| } | |
| /** | |
| * Returns the parent Artifact containing this Artifact. Artifacts without parents shall return | |
| * null. | |
| */ | |
| @Nullable | |
| public SpecialArtifact getParent() { | |
| return null; | |
| } | |
| /** | |
| * Returns the directory name of this artifact, similar to dirname(1). | |
| * | |
| * <p> The directory name is always a relative path to the execution directory. | |
| */ | |
| @Override | |
| public final String getDirname() { | |
| PathFragment parent = getExecPath().getParentDirectory(); | |
| return (parent == null) ? "/" : parent.getSafePathString(); | |
| } | |
| /** | |
| * Returns the base file name of this artifact, similar to basename(1). | |
| */ | |
| @Override | |
| public final String getFilename() { | |
| return getExecPath().getBaseName(); | |
| } | |
| @Override | |
| public final String getExtension() { | |
| return getExecPath().getFileExtension(); | |
| } | |
| /** | |
| * Checks whether this artifact is of the supplied file type. | |
| * | |
| * <p>Prefer this method to pulling out strings from the Artifact and passing to {@link | |
| * FileType#matches(String)} manually. This method has been optimized to generate a minimum of | |
| * garbage. | |
| */ | |
| public boolean isFileType(FileType fileType) { | |
| return fileType.matches(this); | |
| } | |
| @Override | |
| public String filePathForFileTypeMatcher() { | |
| return getExecPath().filePathForFileTypeMatcher(); | |
| } | |
| @Override | |
| public String expandToCommandLine() { | |
| return getExecPathString(); | |
| } | |
| /** Returns the artifact's owning label. May be null. */ | |
| @Nullable | |
| public final Label getOwner() { | |
| return getOwnerLabel(); | |
| } | |
| /** | |
| * Gets the {@code ActionLookupKey} of the {@code ConfiguredTarget} that owns this artifact, if it | |
| * was set. Otherwise, this should be a dummy value -- either {@link | |
| * ArtifactOwner.NullArtifactOwner#INSTANCE} or a dummy owner set in tests. Such a dummy value | |
| * should only occur for source artifacts if created without specifying the owner, or for special | |
| * derived artifacts, such as target completion middleman artifacts, build info artifacts, and the | |
| * like. | |
| */ | |
| public abstract ArtifactOwner getArtifactOwner(); | |
| /** | |
| * Returns the root beneath which this Artifact resides, if any. This may be one of the | |
| * package-path entries (for source Artifacts), or one of the bin, genfiles or includes dirs (for | |
| * derived Artifacts). It will always be an ancestor of getPath(). | |
| */ | |
| @Override | |
| public final ArtifactRoot getRoot() { | |
| return root; | |
| } | |
| @Override | |
| public final PathFragment getExecPath() { | |
| return execPath; | |
| } | |
| @Override | |
| public boolean contentBasedPath() { | |
| return contentBasedPath; | |
| } | |
| @Override | |
| public boolean isSymlink() { | |
| return false; | |
| } | |
| /** | |
| * Returns the path of this Artifact relative to this containing Artifact. Since | |
| * ordinary Artifacts correspond to only one Artifact -- itself -- for ordinary Artifacts, | |
| * this just returns the empty path. For special Artifacts, throws | |
| * {@link UnsupportedOperationException}. See also {@link Artifact#getParentRelativePath()}. | |
| */ | |
| public PathFragment getParentRelativePath() { | |
| return PathFragment.EMPTY_FRAGMENT; | |
| } | |
| /** | |
| * Returns true iff this is a source Artifact as determined by its path and root relationships. | |
| * Note that this will report all Artifacts in the output tree, including in the include symlink | |
| * tree, as non-source. | |
| * | |
| * <p>An {@link Artifact} is a {@link SourceArtifact} iff this returns true, and a {@link | |
| * DerivedArtifact} otherwise. | |
| */ | |
| @Override | |
| public final boolean isSourceArtifact() { | |
| return root.isSourceRoot(); | |
| } | |
| /** | |
| * Returns true iff this is a middleman Artifact as determined by its root. | |
| * | |
| * <p>If true, this artifact is necessarily a {@link DerivedArtifact}. | |
| */ | |
| public final boolean isMiddlemanArtifact() { | |
| return getRoot().isMiddlemanRoot(); | |
| } | |
| /** | |
| * Returns true iff this is a TreeArtifact representing a directory tree containing Artifacts. | |
| * | |
| * <p>if true, this artifact is necessarily a {@link SpecialArtifact} with type {@link | |
| * SpecialArtifactType#TREE}. | |
| */ | |
| public boolean isTreeArtifact() { | |
| return false; | |
| } | |
| /** | |
| * Returns whether the artifact represents a Fileset. | |
| * | |
| * <p>if true, this artifact is necessarily a {@link SpecialArtifact} with type {@link | |
| * SpecialArtifactType#FILESET}. | |
| */ | |
| public boolean isFileset() { | |
| return false; | |
| } | |
| /** The disjunction of {@link #isTreeArtifact} and {@link #isFileset}. */ | |
| @Override | |
| public boolean isDirectory() { | |
| return isTreeArtifact() || isFileset(); | |
| } | |
| /** | |
| * Returns true iff metadata cache must return constant metadata for the given artifact. | |
| * | |
| * <p>If true, this artifact is necessarily a {@link SpecialArtifact} with type {@link | |
| * SpecialArtifactType#CONSTANT_METADATA}. | |
| */ | |
| public boolean isConstantMetadata() { | |
| return false; | |
| } | |
| /** {@link Artifact#isSourceArtifact() is true. | |
| * | |
| * <p>Source artifacts have the property that unlike for output artifacts, direct file system | |
| * access for their contents should be safe, even in a distributed context. | |
| * | |
| * TODO(shahan): move {@link Artifact#getPath} to this subclass. | |
| * */ | |
| public static final class SourceArtifact extends Artifact { | |
| private final ArtifactOwner owner; | |
| @VisibleForTesting | |
| public SourceArtifact(ArtifactRoot root, PathFragment execPath, ArtifactOwner owner) { | |
| super(root, execPath, /*contentBasedPath=*/ false); | |
| this.owner = owner; | |
| } | |
| /** | |
| * Source artifacts do not consider their owners in equality checks, since their owners are | |
| * purely cosmetic. | |
| */ | |
| @Override | |
| boolean ownersEqual(Artifact other) { | |
| return true; | |
| } | |
| @Override | |
| public PathFragment getRootRelativePath() { | |
| // flag-less way of checking of the root is <execroot>/.., or sibling of __main__. | |
| if (getExecPath().startsWith(LabelConstants.EXPERIMENTAL_EXTERNAL_PATH_PREFIX)) { | |
| return LabelConstants.EXTERNAL_PATH_PREFIX.getRelative(getExecPath().subFragment(1)); | |
| } | |
| return getExecPath(); | |
| } | |
| @Override | |
| public ArtifactOwner getArtifactOwner() { | |
| return owner; | |
| } | |
| @Override | |
| public Label getOwnerLabel() { | |
| return owner.getLabel(); | |
| } | |
| boolean differentOwnerOrRoot(ArtifactOwner owner, ArtifactRoot root) { | |
| return !this.owner.equals(owner) || !this.getRoot().equals(root); | |
| } | |
| } | |
| /** | |
| * Special artifact types. | |
| * | |
| * @see SpecialArtifact | |
| */ | |
| @VisibleForTesting | |
| public enum SpecialArtifactType { | |
| /** Google-specific legacy type. */ | |
| FILESET, | |
| /** | |
| * A symlink. Not chased, can be dangling. All we care about is the return value of {@code | |
| * readlink()}. | |
| */ | |
| UNRESOLVED_SYMLINK, | |
| /** A subtree containing multiple files and directories. */ | |
| TREE, | |
| /** Special artifact type for workspace status information. */ | |
| CONSTANT_METADATA, | |
| } | |
| /** | |
| * A special kind of artifact that either is a fileset or needs special metadata caching behavior. | |
| * | |
| * <p>We subclass {@link DerivedArtifact} instead of storing the special attributes inside in | |
| * order to save memory. The proportion of artifacts that are special is very small, and by not | |
| * having to keep around the attribute for the rest we save some memory. | |
| */ | |
| @Immutable | |
| @AutoCodec | |
| public static final class SpecialArtifact extends DerivedArtifact { | |
| private final SpecialArtifactType type; | |
| @VisibleForTesting | |
| public SpecialArtifact( | |
| ArtifactRoot root, PathFragment execPath, ActionLookupKey owner, SpecialArtifactType type) { | |
| super(root, execPath, owner, /*contentBasedPath=*/ false); | |
| this.type = type; | |
| } | |
| @AutoCodec.VisibleForSerialization | |
| @AutoCodec.Instantiator | |
| static SpecialArtifact create( | |
| ArtifactRoot root, | |
| PathFragment execPath, | |
| SpecialArtifactType type, | |
| ActionLookupData generatingActionKey) { | |
| SpecialArtifact result = | |
| new SpecialArtifact(root, execPath, generatingActionKey.getActionLookupKey(), type); | |
| result.setGeneratingActionKey(generatingActionKey); | |
| return result; | |
| } | |
| @Override | |
| public final boolean isFileset() { | |
| return type == SpecialArtifactType.FILESET; | |
| } | |
| @Override | |
| public boolean isConstantMetadata() { | |
| return type == SpecialArtifactType.CONSTANT_METADATA; | |
| } | |
| @Override | |
| public boolean isTreeArtifact() { | |
| return type == SpecialArtifactType.TREE; | |
| } | |
| @Override | |
| public boolean isSymlink() { | |
| return type == SpecialArtifactType.UNRESOLVED_SYMLINK; | |
| } | |
| @Override | |
| public boolean hasParent() { | |
| return false; | |
| } | |
| @Override | |
| @Nullable | |
| public SpecialArtifact getParent() { | |
| return null; | |
| } | |
| @Override | |
| @Nullable | |
| public PathFragment getParentRelativePath() { | |
| return null; | |
| } | |
| @Override | |
| public ShareabilityOfValue getShareabilityOfValue() { | |
| return isConstantMetadata() ? ShareabilityOfValue.NEVER : super.getShareabilityOfValue(); | |
| } | |
| } | |
| /** | |
| * A special kind of artifact that represents a concrete file created at execution time under its | |
| * associated TreeArtifact. | |
| * | |
| * <p>TreeFileArtifacts should be only created during execution time inside some special actions | |
| * to support action inputs and outputs that are unpredictable at analysis time. TreeFileArtifacts | |
| * should not be created directly by any rules at analysis time. | |
| * | |
| * <p>We subclass {@link DerivedArtifact} instead of storing the extra fields directly inside in | |
| * order to save memory. The proportion of TreeFileArtifacts is very small, and by not having to | |
| * keep around the extra fields for the rest we save some memory. | |
| */ | |
| @Immutable | |
| @AutoCodec | |
| public static final class TreeFileArtifact extends DerivedArtifact { | |
| private final SpecialArtifact parentTreeArtifact; | |
| private final PathFragment parentRelativePath; | |
| /** | |
| * Constructs a TreeFileArtifact with the given parent-relative path under the given parent | |
| * TreeArtifact. The {@link ArtifactOwner} of the TreeFileArtifact is the {@link ArtifactOwner} | |
| * of the parent TreeArtifact. | |
| */ | |
| @VisibleForTesting | |
| public TreeFileArtifact(SpecialArtifact parent, PathFragment parentRelativePath) { | |
| this(parent, parentRelativePath, parent.getArtifactOwner()); | |
| } | |
| /** | |
| * Constructs a TreeFileArtifact with the given parent-relative path under the given parent | |
| * TreeArtifact, owned by the given {@code artifactOwner}. | |
| */ | |
| TreeFileArtifact( | |
| SpecialArtifact parentTreeArtifact, | |
| PathFragment parentRelativePath, | |
| ActionLookupKey owner) { | |
| super( | |
| parentTreeArtifact.getRoot(), | |
| parentTreeArtifact.getExecPath().getRelative(parentRelativePath), | |
| owner, | |
| /*contentBasedPath=*/ false); | |
| Preconditions.checkArgument( | |
| parentTreeArtifact.isTreeArtifact(), | |
| "The parent of TreeFileArtifact (parent-relative path: %s) is not a TreeArtifact: %s", | |
| parentRelativePath, | |
| parentTreeArtifact); | |
| Preconditions.checkArgument( | |
| !parentRelativePath.containsUplevelReferences() && !parentRelativePath.isAbsolute(), | |
| "%s is not a proper normalized relative path", | |
| parentRelativePath); | |
| Preconditions.checkState( | |
| parentTreeArtifact.isTreeArtifact(), | |
| "Given parent %s must be a TreeArtifact", | |
| parentTreeArtifact); | |
| this.parentTreeArtifact = parentTreeArtifact; | |
| this.parentRelativePath = parentRelativePath; | |
| } | |
| @AutoCodec.VisibleForSerialization | |
| @AutoCodec.Instantiator | |
| static TreeFileArtifact createForSerialization( | |
| SpecialArtifact parentTreeArtifact, | |
| PathFragment parentRelativePath, | |
| ActionLookupData generatingActionKey) { | |
| TreeFileArtifact result = | |
| new TreeFileArtifact( | |
| parentTreeArtifact, parentRelativePath, generatingActionKey.getActionLookupKey()); | |
| result.setGeneratingActionKey(generatingActionKey); | |
| return result; | |
| } | |
| @Override | |
| public SpecialArtifact getParent() { | |
| return parentTreeArtifact; | |
| } | |
| @Override | |
| public PathFragment getParentRelativePath() { | |
| return parentRelativePath; | |
| } | |
| } | |
| /** | |
| * Returns the relative path to this artifact relative to its root. (Useful when deriving output | |
| * filenames from input files, etc.) | |
| */ | |
| public abstract PathFragment getRootRelativePath(); | |
| /** | |
| * For targets in external repositories, this returns the path the artifact live at in the | |
| * runfiles tree. For local targets, it returns the rootRelativePath. | |
| */ | |
| public final PathFragment getRunfilesPath() { | |
| PathFragment relativePath = getRootRelativePath(); | |
| if (relativePath.startsWith(LabelConstants.EXTERNAL_PATH_PREFIX)) { | |
| // Turn external/repo/foo into ../repo/foo. | |
| relativePath = relativePath.relativeTo(LabelConstants.EXTERNAL_PATH_PREFIX); | |
| relativePath = PathFragment.create("..").getRelative(relativePath); | |
| } | |
| return relativePath; | |
| } | |
| @Override | |
| public final String getRunfilesPathString() { | |
| return getRunfilesPath().getPathString(); | |
| } | |
| /** | |
| * Returns this.getExecPath().getPathString(). | |
| */ | |
| @Override | |
| public final String getExecPathString() { | |
| return getExecPath().getPathString(); | |
| } | |
| public final String getRootRelativePathString() { | |
| return getRootRelativePath().getPathString(); | |
| } | |
| public final String prettyPrint() { | |
| // toDetailString would probably be more useful to users, but lots of tests rely on the | |
| // current values. | |
| return getRootRelativePath().toString(); | |
| } | |
| @SuppressWarnings("EqualsGetClass") // Distinct classes of Artifact are never equal. | |
| @Override | |
| public final boolean equals(Object other) { | |
| if (this == other) { | |
| return true; | |
| } | |
| if (!(other instanceof Artifact)) { | |
| return false; | |
| } | |
| if (!getClass().equals(other.getClass())) { | |
| return false; | |
| } | |
| Artifact that = (Artifact) other; | |
| return equalsWithoutOwner(that) && ownersEqual(that); | |
| } | |
| final boolean equalsWithoutOwner(Artifact other) { | |
| return hashCode == other.hashCode && execPath.equals(other.execPath) && root.equals(other.root); | |
| } | |
| abstract boolean ownersEqual(Artifact other); | |
| @Override | |
| public final int hashCode() { | |
| // This is just execPath.hashCode() (along with the class). We cache a copy in the Artifact | |
| // object to reduce LLC misses during operations which build a HashSet out of many Artifacts. | |
| // This is a slight loss for memory but saves ~1% overall CPU in some real builds. | |
| return hashCode; | |
| } | |
| @Override | |
| public final String toString() { | |
| return "File:" + toDetailString(); | |
| } | |
| /** | |
| * Returns a string representing the complete artifact path information. | |
| */ | |
| public final String toDetailString() { | |
| if (isSourceArtifact()) { | |
| // Source Artifact: relPath == execPath, & real path is not under execRoot | |
| return "[" + root + "]" + getRootRelativePathString(); | |
| } else { | |
| // Derived Artifact: path and root are under execRoot | |
| // | |
| // TODO(blaze-team): this is misleading because execution_root isn't unique. Dig the | |
| // workspace name out and print that also. | |
| return "[[<execution_root>]" + root.getExecPath() + "]" + getRootRelativePathString(); | |
| } | |
| } | |
| /** {@link ObjectCodec} for {@link SourceArtifact} */ | |
| @SuppressWarnings("unused") // found by CLASSPATH-scanning magic | |
| private static class SourceArtifactCodec implements ObjectCodec<SourceArtifact> { | |
| @Override | |
| public Class<? extends SourceArtifact> getEncodedClass() { | |
| return SourceArtifact.class; | |
| } | |
| @Override | |
| public void serialize( | |
| SerializationContext context, SourceArtifact obj, CodedOutputStream codedOut) | |
| throws SerializationException, IOException { | |
| context.serialize(obj.getExecPath(), codedOut); | |
| context.serialize(obj.getRoot(), codedOut); | |
| context.serialize(obj.getArtifactOwner(), codedOut); | |
| } | |
| @Override | |
| public SourceArtifact deserialize(DeserializationContext context, CodedInputStream codedIn) | |
| throws SerializationException, IOException { | |
| PathFragment execPath = context.deserialize(codedIn); | |
| ArtifactRoot artifactRoot = context.deserialize(codedIn); | |
| ArtifactOwner owner = context.deserialize(codedIn); | |
| return (SourceArtifact) | |
| context | |
| .getDependency(ArtifactResolverSupplier.class) | |
| .get() | |
| .getSourceArtifact(execPath, artifactRoot.getRoot(), owner); | |
| } | |
| } | |
| // --------------------------------------------------------------------------- | |
| // Static methods to assist in working with Artifacts | |
| /** Formatter for execPath PathFragment output. */ | |
| public static final Function<Artifact, String> ROOT_RELATIVE_PATH_STRING = | |
| artifact -> artifact.getRootRelativePath().getPathString(); | |
| /** | |
| * Converts a collection of artifacts into execution-time path strings, and | |
| * adds those to a given collection. Middleman artifacts are ignored by this | |
| * method. | |
| */ | |
| public static void addExecPaths(Iterable<Artifact> artifacts, Collection<String> output) { | |
| addNonMiddlemanArtifacts(artifacts, output, ActionInputHelper.EXEC_PATH_STRING_FORMATTER); | |
| } | |
| /** | |
| * Converts a collection of artifacts into the outputs computed by | |
| * outputFormatter and adds them to a given collection. Middleman artifacts | |
| * are ignored. | |
| */ | |
| static <E> void addNonMiddlemanArtifacts(Iterable<Artifact> artifacts, | |
| Collection<? super E> output, Function<? super Artifact, E> outputFormatter) { | |
| for (Artifact artifact : artifacts) { | |
| if (MIDDLEMAN_FILTER.apply(artifact)) { | |
| output.add(outputFormatter.apply(artifact)); | |
| } | |
| } | |
| } | |
| /** | |
| * Lazily converts artifacts into root-relative path strings. Middleman artifacts are ignored by | |
| * this method. | |
| */ | |
| public static Iterable<String> toRootRelativePaths(NestedSet<Artifact> artifacts) { | |
| return toRootRelativePaths(artifacts.toList()); | |
| } | |
| /** | |
| * Lazily converts artifacts into root-relative path strings. Middleman artifacts are ignored by | |
| * this method. | |
| */ | |
| public static Iterable<String> toRootRelativePaths(Iterable<Artifact> artifacts) { | |
| return Iterables.transform( | |
| Iterables.filter(artifacts, MIDDLEMAN_FILTER), | |
| artifact -> artifact.getRootRelativePath().getPathString()); | |
| } | |
| /** | |
| * Lazily converts artifacts into execution-time path strings. Middleman artifacts are ignored by | |
| * this method. | |
| */ | |
| public static Iterable<String> toExecPaths(Iterable<Artifact> artifacts) { | |
| return ActionInputHelper.toExecPaths(Iterables.filter(artifacts, MIDDLEMAN_FILTER)); | |
| } | |
| /** | |
| * Converts a collection of artifacts into execution-time path strings, and returns those as an | |
| * immutable list. Middleman artifacts are ignored by this method. | |
| * | |
| * <p>Avoid this method in production code - it flattens the given nested set unconditionally. | |
| */ | |
| @VisibleForTesting | |
| public static List<String> asExecPaths(NestedSet<Artifact> artifacts) { | |
| return asExecPaths(artifacts.toList()); | |
| } | |
| /** | |
| * Converts a collection of artifacts into execution-time path strings, and | |
| * returns those as an immutable list. Middleman artifacts are ignored by this method. | |
| */ | |
| public static List<String> asExecPaths(Iterable<Artifact> artifacts) { | |
| return ImmutableList.copyOf(toExecPaths(artifacts)); | |
| } | |
| /** | |
| * Renders a collection of artifacts as execution-time paths and joins | |
| * them into a single string. Middleman artifacts are ignored by this method. | |
| */ | |
| public static String joinExecPaths(String delimiter, Iterable<Artifact> artifacts) { | |
| return Joiner.on(delimiter).join(toExecPaths(artifacts)); | |
| } | |
| /** | |
| * Renders a collection of artifacts as root-relative paths and joins | |
| * them into a single string. Middleman artifacts are ignored by this method. | |
| */ | |
| public static String joinRootRelativePaths(String delimiter, Iterable<Artifact> artifacts) { | |
| return Joiner.on(delimiter).join(toRootRelativePaths(artifacts)); | |
| } | |
| /** | |
| * Adds a collection of artifacts to a given collection, with | |
| * {@link MiddlemanType#AGGREGATING_MIDDLEMAN} middleman actions expanded once. | |
| */ | |
| public static void addExpandedArtifacts(Iterable<Artifact> artifacts, | |
| Collection<? super Artifact> output, ArtifactExpander artifactExpander) { | |
| addExpandedArtifacts(artifacts, output, Functions.<Artifact>identity(), artifactExpander); | |
| } | |
| /** | |
| * Converts a collection of artifacts into the outputs computed by | |
| * outputFormatter and adds them to a given collection. Middleman artifacts | |
| * are expanded once. | |
| */ | |
| private static <E> void addExpandedArtifacts(Iterable<? extends Artifact> artifacts, | |
| Collection<? super E> output, | |
| Function<? super Artifact, E> outputFormatter, | |
| ArtifactExpander artifactExpander) { | |
| for (Artifact artifact : artifacts) { | |
| if (artifact.isMiddlemanArtifact() || artifact.isTreeArtifact()) { | |
| expandArtifact(artifact, output, outputFormatter, artifactExpander); | |
| } else { | |
| output.add(outputFormatter.apply(artifact)); | |
| } | |
| } | |
| } | |
| private static <E> void expandArtifact(Artifact middleman, | |
| Collection<? super E> output, | |
| Function<? super Artifact, E> outputFormatter, | |
| ArtifactExpander artifactExpander) { | |
| Preconditions.checkArgument(middleman.isMiddlemanArtifact() || middleman.isTreeArtifact()); | |
| List<Artifact> artifacts = new ArrayList<>(); | |
| artifactExpander.expand(middleman, artifacts); | |
| for (Artifact artifact : artifacts) { | |
| output.add(outputFormatter.apply(artifact)); | |
| } | |
| } | |
| /** | |
| * Convenience method to filter the files to build for a certain filetype. | |
| * | |
| * @param artifacts the files to filter | |
| * @param allowedType the allowed filetype | |
| * @return all members of filesToBuild that are of one of the | |
| * allowed filetypes | |
| */ | |
| public static List<Artifact> filterFiles(Iterable<Artifact> artifacts, FileType allowedType) { | |
| List<Artifact> filesToBuild = new ArrayList<>(); | |
| for (Artifact artifact : artifacts) { | |
| if (allowedType.matches(artifact.getFilename())) { | |
| filesToBuild.add(artifact); | |
| } | |
| } | |
| return filesToBuild; | |
| } | |
| /** | |
| * Converts artifacts into their exec paths. Returns an immutable list. | |
| */ | |
| public static List<PathFragment> asPathFragments(Iterable<? extends Artifact> artifacts) { | |
| return Streams.stream(artifacts).map(Artifact::getExecPath).collect(toImmutableList()); | |
| } | |
| /** | |
| * Returns the exec paths of the input artifacts in alphabetical order. | |
| */ | |
| public static ImmutableList<PathFragment> asSortedPathFragments(Iterable<Artifact> input) { | |
| return Streams.stream(input).map(Artifact::getExecPath).sorted().collect(toImmutableList()); | |
| } | |
| @Override | |
| public boolean isImmutable() { | |
| return true; | |
| } | |
| @Override | |
| public void repr(Printer printer) { | |
| if (isSourceArtifact()) { | |
| printer.append("<source file " + getRootRelativePathString() + ">"); | |
| } else { | |
| printer.append("<generated file " + getRootRelativePathString() + ">"); | |
| } | |
| } | |
| /** | |
| * A utility class that compares {@link Artifact}s without taking their owners into account. | |
| * Should only be used for detecting action conflicts and merging shared action data. | |
| */ | |
| public static class OwnerlessArtifactWrapper { | |
| private final Artifact artifact; | |
| public OwnerlessArtifactWrapper(Artifact artifact) { | |
| this.artifact = artifact; | |
| } | |
| @Override | |
| public int hashCode() { | |
| // Depends on the fact that Artifact#hashCode does not use ArtifactOwner. | |
| return artifact.hashCode(); | |
| } | |
| @Override | |
| public boolean equals(Object obj) { | |
| return obj instanceof OwnerlessArtifactWrapper | |
| && this.artifact.equalsWithoutOwner(((OwnerlessArtifactWrapper) obj).artifact); | |
| } | |
| } | |
| @Override | |
| public SkyFunctionName functionName() { | |
| return ARTIFACT; | |
| } | |
| } |