| // 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.skyframe; |
| |
| import static com.google.devtools.build.lib.vfs.UnixGlob.DEFAULT_SYSCALLS; |
| import static java.nio.charset.StandardCharsets.UTF_8; |
| |
| import com.google.common.annotations.VisibleForTesting; |
| import com.google.common.base.Preconditions; |
| import com.google.common.base.Verify; |
| import com.google.common.collect.Collections2; |
| import com.google.common.collect.ImmutableList; |
| import com.google.devtools.build.lib.actions.Artifact; |
| import com.google.devtools.build.lib.actions.Artifact.TreeFileArtifact; |
| import com.google.devtools.build.lib.actions.FileArtifactValue; |
| import com.google.devtools.build.lib.actions.FileStateType; |
| import com.google.devtools.build.lib.actions.FileStateValue; |
| import com.google.devtools.build.lib.actions.FileStateValue.RegularFileStateValue; |
| import com.google.devtools.build.lib.actions.FileValue; |
| import com.google.devtools.build.lib.actions.HasDigest; |
| import com.google.devtools.build.lib.collect.nestedset.NestedSet; |
| import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder; |
| import com.google.devtools.build.lib.events.Event; |
| import com.google.devtools.build.lib.skyframe.RecursiveFilesystemTraversalValue.FileType; |
| import com.google.devtools.build.lib.skyframe.RecursiveFilesystemTraversalValue.ResolvedFile; |
| import com.google.devtools.build.lib.skyframe.RecursiveFilesystemTraversalValue.ResolvedFileFactory; |
| import com.google.devtools.build.lib.skyframe.RecursiveFilesystemTraversalValue.TraversalRequest; |
| import com.google.devtools.build.lib.skyframe.serialization.autocodec.AutoCodec; |
| import com.google.devtools.build.lib.util.Fingerprint; |
| import com.google.devtools.build.lib.vfs.Dirent; |
| import com.google.devtools.build.lib.vfs.FileStatus; |
| import com.google.devtools.build.lib.vfs.Path; |
| 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.lib.vfs.Symlinks; |
| import com.google.devtools.build.skyframe.SkyFunction; |
| import com.google.devtools.build.skyframe.SkyFunctionException; |
| import com.google.devtools.build.skyframe.SkyKey; |
| import com.google.devtools.build.skyframe.SkyValue; |
| import java.io.IOException; |
| import java.math.BigInteger; |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| import javax.annotation.Nullable; |
| |
| /** A {@link SkyFunction} to build {@link RecursiveFilesystemTraversalValue}s. */ |
| public final class RecursiveFilesystemTraversalFunction implements SkyFunction { |
| |
| private static final class MissingDepException extends Exception {} |
| |
| private static final byte[] MISSING_FINGERPRINT = |
| new BigInteger(1, "NonexistentFileStateValue".getBytes(UTF_8)).toByteArray(); |
| |
| @AutoCodec @AutoCodec.VisibleForSerialization |
| static final HasDigest NON_EXISTENT_HAS_DIGEST = () -> MISSING_FINGERPRINT; |
| |
| private static final FileInfo NON_EXISTENT_FILE_INFO = |
| new FileInfo(FileType.NONEXISTENT, NON_EXISTENT_HAS_DIGEST, null, null); |
| |
| /** The exception that {@link RecursiveFilesystemTraversalFunctionException} wraps. */ |
| public static class RecursiveFilesystemTraversalException extends Exception { |
| |
| /** |
| * Categories of errors that prevent normal {@link RecursiveFilesystemTraversalFunction} |
| * evaluation. |
| */ |
| public enum Type { |
| /** |
| * The traversal encountered a subdirectory with a BUILD file but is not allowed to recurse |
| * into it. See {@code PackageBoundaryMode#REPORT_ERROR}. |
| */ |
| CANNOT_CROSS_PACKAGE_BOUNDARY, |
| |
| /** A dangling symlink was dereferenced. */ |
| DANGLING_SYMLINK, |
| |
| /** A file operation failed. */ |
| FILE_OPERATION_FAILURE, |
| |
| /** A generated directory's root-relative path conflicts with a package's path. */ |
| GENERATED_PATH_CONFLICT, |
| } |
| |
| private final Type type; |
| |
| RecursiveFilesystemTraversalException(String message, Type type) { |
| super(message); |
| this.type = type; |
| } |
| |
| public Type getType() { |
| return type; |
| } |
| } |
| |
| /** |
| * Thrown when a dangling symlink is attempted to be dereferenced. |
| * |
| * <p>Note: this class is not identical to the one in com.google.devtools.build.lib.view.fileset |
| * and it's not easy to merge the two because of the dependency structure. The other one will |
| * probably be removed along with the rest of the legacy Fileset code. |
| */ |
| static final class DanglingSymlinkException extends RecursiveFilesystemTraversalException { |
| DanglingSymlinkException(String path, String unresolvedLink) { |
| super( |
| String.format( |
| "Found dangling symlink: %s, unresolved path: \"%s\"", path, unresolvedLink), |
| Type.DANGLING_SYMLINK); |
| Preconditions.checkArgument(path != null && !path.isEmpty()); |
| Preconditions.checkArgument(unresolvedLink != null && !unresolvedLink.isEmpty()); |
| } |
| } |
| |
| /** Exception type thrown by {@link RecursiveFilesystemTraversalFunction#compute}. */ |
| private static final class RecursiveFilesystemTraversalFunctionException extends |
| SkyFunctionException { |
| RecursiveFilesystemTraversalFunctionException(RecursiveFilesystemTraversalException e) { |
| super(e, Transience.PERSISTENT); |
| } |
| } |
| |
| @Override |
| public SkyValue compute(SkyKey skyKey, Environment env) |
| throws RecursiveFilesystemTraversalFunctionException, InterruptedException { |
| TraversalRequest traversal = (TraversalRequest) skyKey.argument(); |
| try { |
| // Stat the traversal root. |
| FileInfo rootInfo = lookUpFileInfo(env, traversal); |
| |
| if (!rootInfo.type.exists()) { |
| // May be a dangling symlink or a non-existent file. Handle gracefully. |
| if (rootInfo.type.isSymlink()) { |
| return resultForDanglingSymlink(traversal.root.asRootedPath(), rootInfo); |
| } else { |
| return RecursiveFilesystemTraversalValue.EMPTY; |
| } |
| } |
| |
| if (rootInfo.type.isFile()) { |
| return resultForFileRoot(traversal.root.asRootedPath(), rootInfo); |
| } else if (rootInfo.type.isDirectory() && rootInfo.metadata instanceof TreeArtifactValue) { |
| final TreeArtifactValue value = (TreeArtifactValue) rootInfo.metadata; |
| ImmutableList.Builder<RecursiveFilesystemTraversalValue> list = ImmutableList.builder(); |
| for (Map.Entry<TreeFileArtifact, FileArtifactValue> entry |
| : value.getChildValues().entrySet()) { |
| RootedPath path = |
| RootedPath.toRootedPath( |
| traversal.root.asRootedPath().getRoot(), entry.getKey().getPath()); |
| list.add( |
| resultForFileRoot( |
| path, |
| // TreeArtifact can't have symbolic inside. So the assumption for FileType.FILE |
| // is always true. |
| new FileInfo(FileType.FILE, entry.getValue(), path, null))); |
| } |
| return resultForDirectory(traversal, rootInfo, list.build()); |
| } |
| |
| // Otherwise the root is a directory or a symlink to one. |
| PkgLookupResult pkgLookupResult = checkIfPackage(env, traversal, rootInfo); |
| traversal = pkgLookupResult.traversal; |
| |
| if (pkgLookupResult.isConflicting()) { |
| // The traversal was requested for an output directory whose root-relative path conflicts |
| // with a source package. We can't handle that, bail out. |
| throw createGeneratedPathConflictException(traversal); |
| } else if (pkgLookupResult.isPackage() && !traversal.skipTestingForSubpackage) { |
| // The traversal was requested for a directory that defines a package. |
| String msg = |
| traversal.errorInfo |
| + " crosses package boundary into package rooted at " |
| + traversal.root.asRootedPath().getRootRelativePath().getPathString(); |
| switch (traversal.crossPkgBoundaries) { |
| case CROSS: |
| // We are free to traverse the subpackage but we need to display a warning. |
| env.getListener().handle(Event.warn(null, msg)); |
| break; |
| case DONT_CROSS: |
| // We cannot traverse the subpackage and should skip it silently. Return empty results. |
| return RecursiveFilesystemTraversalValue.EMPTY; |
| case REPORT_ERROR: |
| // We cannot traverse the subpackage and should complain loudly (display an error). |
| throw new RecursiveFilesystemTraversalFunctionException( |
| new RecursiveFilesystemTraversalException( |
| msg, RecursiveFilesystemTraversalException.Type.CANNOT_CROSS_PACKAGE_BOUNDARY)); |
| default: |
| throw new IllegalStateException(traversal.toString()); |
| } |
| } |
| |
| // We are free to traverse this directory. |
| Collection<SkyKey> dependentKeys = createRecursiveTraversalKeys(env, traversal); |
| return resultForDirectory( |
| traversal, |
| rootInfo, |
| traverseChildren(env, dependentKeys, /*inline=*/ traversal.isRootGenerated)); |
| } catch (IOException e) { |
| String message = "Error while traversing fileset: " + e.getMessage(); |
| throw new RecursiveFilesystemTraversalFunctionException( |
| new RecursiveFilesystemTraversalException( |
| message, RecursiveFilesystemTraversalException.Type.FILE_OPERATION_FAILURE)); |
| } catch (MissingDepException e) { |
| return null; |
| } |
| } |
| |
| private static RecursiveFilesystemTraversalFunctionException createGeneratedPathConflictException( |
| TraversalRequest traversal) { |
| String message = |
| String.format( |
| "Generated directory %s conflicts with package under the same path. " |
| + "Additional info: %s", |
| traversal.root.asRootedPath().getRootRelativePath().getPathString(), |
| traversal.errorInfo != null ? traversal.errorInfo : traversal.toString()); |
| return new RecursiveFilesystemTraversalFunctionException( |
| new RecursiveFilesystemTraversalException( |
| message, RecursiveFilesystemTraversalException.Type.GENERATED_PATH_CONFLICT)); |
| } |
| |
| @Override |
| public String extractTag(SkyKey skyKey) { |
| return null; |
| } |
| |
| private static final class FileInfo { |
| final FileType type; |
| final HasDigest metadata; |
| @Nullable final RootedPath realPath; |
| @Nullable final PathFragment unresolvedSymlinkTarget; |
| |
| FileInfo( |
| FileType type, |
| HasDigest metadata, |
| @Nullable RootedPath realPath, |
| @Nullable PathFragment unresolvedSymlinkTarget) { |
| Preconditions.checkNotNull(metadata.getDigest(), metadata); |
| this.type = Preconditions.checkNotNull(type); |
| this.metadata = metadata; |
| this.realPath = realPath; |
| this.unresolvedSymlinkTarget = unresolvedSymlinkTarget; |
| } |
| |
| @Override |
| public String toString() { |
| if (type.isSymlink()) { |
| return String.format("(%s: link_value=%s, real_path=%s)", type, |
| unresolvedSymlinkTarget.getPathString(), realPath); |
| } else { |
| return String.format("(%s: real_path=%s)", type, realPath); |
| } |
| } |
| } |
| |
| private static FileInfo lookUpFileInfo(Environment env, TraversalRequest traversal) |
| throws MissingDepException, IOException, InterruptedException { |
| if (traversal.isRootGenerated) { |
| HasDigest fsVal = null; |
| if (traversal.root.getOutputArtifact() != null) { |
| Artifact artifact = traversal.root.getOutputArtifact(); |
| SkyKey artifactKey = Artifact.key(artifact); |
| SkyValue value = env.getValue(artifactKey); |
| if (env.valuesMissing()) { |
| throw new MissingDepException(); |
| } |
| |
| if (value instanceof FileArtifactValue || value instanceof TreeArtifactValue) { |
| fsVal = (HasDigest) value; |
| } else if (value instanceof ActionExecutionValue) { |
| fsVal = ((ActionExecutionValue) value).getExistingFileArtifactValue(artifact); |
| } else { |
| return NON_EXISTENT_FILE_INFO; |
| } |
| } |
| RootedPath realPath = traversal.root.asRootedPath(); |
| if (traversal.strictOutputFiles) { |
| Preconditions.checkNotNull(fsVal, "Strict Fileset output tree has null FileArtifactValue"); |
| return new FileInfo( |
| (fsVal instanceof TreeArtifactValue ? FileType.DIRECTORY : FileType.FILE), |
| fsVal, |
| realPath, |
| null); |
| } else { |
| // FileArtifactValue does not currently track symlinks. If it did, we could potentially |
| // remove some of the filesystem operations we're doing here. |
| Path path = traversal.root.asRootedPath().asPath(); |
| FileStateValue fileState = |
| FileStateValue.create(traversal.root.asRootedPath(), DEFAULT_SYSCALLS, null); |
| if (fileState.getType() == FileStateType.NONEXISTENT) { |
| throw new IOException("Missing file: " + path); |
| } |
| FileStatus followStat = path.statIfFound(Symlinks.FOLLOW); |
| FileType type; |
| PathFragment unresolvedLinkTarget = null; |
| if (followStat == null) { |
| type = FileType.DANGLING_SYMLINK; |
| if (fileState.getType() != FileStateType.SYMLINK) { |
| throw new IOException("Expected symlink for " + path + ", but got: " + fileState); |
| } |
| unresolvedLinkTarget = path.readSymbolicLink(); |
| } else if (fileState.getType() == FileStateType.REGULAR_FILE) { |
| type = FileType.FILE; |
| } else if (fileState.getType() == FileStateType.DIRECTORY) { |
| type = FileType.DIRECTORY; |
| } else { |
| unresolvedLinkTarget = path.readSymbolicLink(); |
| realPath = |
| RootedPath.toRootedPath( |
| Root.absoluteRoot(path.getFileSystem()), path.resolveSymbolicLinks()); |
| type = followStat.isFile() ? FileType.SYMLINK_TO_FILE : FileType.SYMLINK_TO_DIRECTORY; |
| } |
| if (fsVal == null) { |
| fsVal = fileState; |
| } |
| return new FileInfo(type, withDigest(fsVal, path), realPath, unresolvedLinkTarget); |
| } |
| } else { |
| // Stat the file. |
| FileValue fileValue = |
| (FileValue) env.getValueOrThrow( |
| FileValue.key(traversal.root.asRootedPath()), IOException.class); |
| |
| if (env.valuesMissing()) { |
| throw new MissingDepException(); |
| } |
| if (fileValue.exists()) { |
| // If it exists, it may either be a symlink or a file/directory. |
| PathFragment unresolvedLinkTarget = null; |
| FileType type; |
| if (fileValue.isSymlink()) { |
| unresolvedLinkTarget = fileValue.getUnresolvedLinkTarget(); |
| type = fileValue.isDirectory() ? FileType.SYMLINK_TO_DIRECTORY : FileType.SYMLINK_TO_FILE; |
| } else { |
| type = fileValue.isDirectory() ? FileType.DIRECTORY : FileType.FILE; |
| } |
| Path path = traversal.root.asRootedPath().asPath(); |
| return new FileInfo( |
| type, |
| withDigest(fileValue.realFileStateValue(), path), |
| fileValue.realRootedPath(), |
| unresolvedLinkTarget); |
| } else { |
| // If it doesn't exist, or it's a dangling symlink, we still want to handle that gracefully. |
| return new FileInfo( |
| fileValue.isSymlink() ? FileType.DANGLING_SYMLINK : FileType.NONEXISTENT, |
| withDigest(fileValue.realFileStateValue(), null), |
| null, |
| fileValue.isSymlink() ? fileValue.getUnresolvedLinkTarget() : null); |
| } |
| } |
| } |
| |
| /** |
| * Transform the HasDigest to the appropriate type based on the current state of the digest. If |
| * fsVal is type RegularFileStateValue or FileArtifactValue and has a valid digest value, then we |
| * want to convert it to a new FileArtifactValue type. Otherwise if they are of the two |
| * forementioned types but do not have a digest, then we will create a FileArtifactValue using its |
| * {@link Path}. Otherwise we will fingerprint the digest and return it as a new {@link |
| * HasDigest.ByteStringDigest} object. |
| * |
| * @param fsVal - the HasDigest value that was in the graph. |
| * @param path - the Path of the digest. |
| * @return transformed HasDigest value based on the digest field and object type. |
| */ |
| @VisibleForTesting |
| static HasDigest withDigest(HasDigest fsVal, Path path) throws IOException { |
| if (fsVal instanceof FileStateValue) { |
| FileStateValue fsv = (FileStateValue) fsVal; |
| if (fsv instanceof RegularFileStateValue) { |
| RegularFileStateValue rfsv = (RegularFileStateValue) fsv; |
| return rfsv.getDigest() != null |
| // If we have the digest, then simply convert it with the digest value. |
| ? FileArtifactValue.createForVirtualActionInput(rfsv.getDigest(), rfsv.getSize()) |
| // Otherwise, create a file FileArtifactValue (RegularFileArtifactValue) based on the |
| // path and size. |
| : FileArtifactValue.createForNormalFileUsingPath(path, rfsv.getSize()); |
| } |
| return new HasDigest.ByteStringDigest(fsv.getValueFingerprint().toByteArray()); |
| } else if (fsVal instanceof FileArtifactValue) { |
| FileArtifactValue fav = ((FileArtifactValue) fsVal); |
| if (fav.getDigest() != null) { |
| return fav; |
| } |
| |
| // In the case there is a directory, the HasDigest value should not be converted. Otherwise, |
| // if the HasDigest value is a file, convert it using the Path and size values. |
| return fav.getType().isFile() |
| ? FileArtifactValue.createForNormalFileUsingPath(path, fav.getSize()) |
| : new HasDigest.ByteStringDigest(fav.getValueFingerprint().toByteArray()); |
| } |
| return fsVal; |
| } |
| |
| private static final class PkgLookupResult { |
| private enum Type { |
| CONFLICT, DIRECTORY, PKG |
| } |
| |
| private final Type type; |
| final TraversalRequest traversal; |
| final FileInfo rootInfo; |
| |
| /** Result for a generated directory that conflicts with a source package. */ |
| static PkgLookupResult conflict(TraversalRequest traversal, FileInfo rootInfo) { |
| return new PkgLookupResult(Type.CONFLICT, traversal, rootInfo); |
| } |
| |
| /** Result for a source or generated directory (not a package). */ |
| static PkgLookupResult directory(TraversalRequest traversal, FileInfo rootInfo) { |
| return new PkgLookupResult(Type.DIRECTORY, traversal, rootInfo); |
| } |
| |
| /** Result for a package, i.e. a directory with a BUILD file. */ |
| static PkgLookupResult pkg(TraversalRequest traversal, FileInfo rootInfo) { |
| return new PkgLookupResult(Type.PKG, traversal, rootInfo); |
| } |
| |
| private PkgLookupResult(Type type, TraversalRequest traversal, FileInfo rootInfo) { |
| this.type = Preconditions.checkNotNull(type); |
| this.traversal = Preconditions.checkNotNull(traversal); |
| this.rootInfo = Preconditions.checkNotNull(rootInfo); |
| } |
| |
| boolean isPackage() { |
| return type == Type.PKG; |
| } |
| |
| boolean isConflicting() { |
| return type == Type.CONFLICT; |
| } |
| |
| @Override |
| public String toString() { |
| return String.format("(%s: info=%s, traversal=%s)", type, rootInfo, traversal); |
| } |
| } |
| |
| /** |
| * Checks whether the {@code traversal}'s path refers to a package directory. |
| * |
| * @return the result of the lookup; it contains potentially new {@link TraversalRequest} and |
| * {@link FileInfo} so the caller should use these instead of the old ones (this happens when |
| * a package is found, but under a different root than expected) |
| */ |
| private static PkgLookupResult checkIfPackage( |
| Environment env, TraversalRequest traversal, FileInfo rootInfo) |
| throws MissingDepException, IOException, InterruptedException { |
| Preconditions.checkArgument(rootInfo.type.exists() && !rootInfo.type.isFile(), |
| "{%s} {%s}", traversal, rootInfo); |
| PackageLookupValue pkgLookup = |
| (PackageLookupValue) |
| getDependentSkyValue(env, |
| PackageLookupValue.key(traversal.root.asRootedPath().getRootRelativePath())); |
| |
| if (pkgLookup.packageExists()) { |
| if (traversal.isRootGenerated) { |
| // The traversal's root was a generated directory, but its root-relative path conflicts with |
| // an existing package. |
| return PkgLookupResult.conflict(traversal, rootInfo); |
| } else { |
| // The traversal's root was a source directory and it defines a package. |
| Root pkgRoot = pkgLookup.getRoot(); |
| if (!pkgRoot.equals(traversal.root.asRootedPath().getRoot())) { |
| // However the root of this package is different from what we expected. stat() the real |
| // BUILD file of that package. |
| traversal = traversal.forChangedRootPath(pkgRoot); |
| rootInfo = lookUpFileInfo(env, traversal); |
| Verify.verify(rootInfo.type.exists(), "{%s} {%s}", traversal, rootInfo); |
| } |
| return PkgLookupResult.pkg(traversal, rootInfo); |
| } |
| } else { |
| // The traversal's root was a directory (source or generated one), no package exists under the |
| // same root-relative path. |
| return PkgLookupResult.directory(traversal, rootInfo); |
| } |
| } |
| |
| /** |
| * List the directory and create {@code SkyKey}s to request contents of its children recursively. |
| * |
| * <p>The returned keys are of type {@link SkyFunctions#RECURSIVE_FILESYSTEM_TRAVERSAL}. |
| */ |
| private static Collection<SkyKey> createRecursiveTraversalKeys( |
| Environment env, TraversalRequest traversal) |
| throws MissingDepException, InterruptedException, IOException { |
| // Use the traversal's path, even if it's a symlink. The contents of the directory, as listed |
| // in the result, must be relative to it. |
| Iterable<Dirent> dirents; |
| if (traversal.isRootGenerated) { |
| // If we're dealing with an output file, read the directory directly instead of creating |
| // filesystem nodes under the output tree. |
| List<Dirent> direntsCollection = |
| new ArrayList<>( |
| traversal.root.asRootedPath().asPath().readdir(Symlinks.FOLLOW)); |
| Collections.sort(direntsCollection); |
| dirents = direntsCollection; |
| } else { |
| dirents = ((DirectoryListingValue) getDependentSkyValue(env, |
| DirectoryListingValue.key(traversal.root.asRootedPath()))).getDirents(); |
| } |
| |
| List<SkyKey> result = new ArrayList<>(); |
| for (Dirent dirent : dirents) { |
| RootedPath childPath = |
| RootedPath.toRootedPath( |
| traversal.root.asRootedPath().getRoot(), |
| traversal.root.asRootedPath().getRootRelativePath().getRelative(dirent.getName())); |
| TraversalRequest childTraversal = traversal.forChildEntry(childPath); |
| result.add(childTraversal); |
| } |
| return result; |
| } |
| |
| /** |
| * Creates result for a dangling symlink. |
| * |
| * @param linkName path to the symbolic link |
| * @param info the {@link FileInfo} associated with the link file |
| */ |
| private static RecursiveFilesystemTraversalValue resultForDanglingSymlink(RootedPath linkName, |
| FileInfo info) { |
| Preconditions.checkState(info.type.isSymlink() && !info.type.exists(), "{%s} {%s}", linkName, |
| info.type); |
| return RecursiveFilesystemTraversalValue.of( |
| ResolvedFileFactory.danglingSymlink(linkName, info.unresolvedSymlinkTarget, info.metadata)); |
| } |
| |
| /** |
| * Creates results for a file or for a symlink that points to one. |
| * |
| * <p>A symlink may be direct (points to a file) or transitive (points at a direct or transitive |
| * symlink). |
| */ |
| private static RecursiveFilesystemTraversalValue resultForFileRoot(RootedPath path, |
| FileInfo info) { |
| Preconditions.checkState(info.type.isFile() && info.type.exists(), "{%s} {%s}", path, |
| info.type); |
| if (info.type.isSymlink()) { |
| return RecursiveFilesystemTraversalValue.of( |
| ResolvedFileFactory.symlinkToFile( |
| info.realPath, path, info.unresolvedSymlinkTarget, info.metadata)); |
| } else { |
| return RecursiveFilesystemTraversalValue.of( |
| ResolvedFileFactory.regularFile(path, info.metadata)); |
| } |
| } |
| |
| private static RecursiveFilesystemTraversalValue resultForDirectory(TraversalRequest traversal, |
| FileInfo rootInfo, Collection<RecursiveFilesystemTraversalValue> subdirTraversals) { |
| // Collect transitive closure of files in subdirectories. |
| NestedSetBuilder<ResolvedFile> paths = NestedSetBuilder.stableOrder(); |
| for (RecursiveFilesystemTraversalValue child : subdirTraversals) { |
| paths.addTransitive(child.getTransitiveFiles()); |
| } |
| ResolvedFile root; |
| if (rootInfo.type.isSymlink()) { |
| NestedSet<ResolvedFile> children = paths.build(); |
| root = |
| ResolvedFileFactory.symlinkToDirectory( |
| rootInfo.realPath, |
| traversal.root.asRootedPath(), |
| rootInfo.unresolvedSymlinkTarget, |
| hashDirectorySymlink(children, rootInfo.metadata)); |
| paths = NestedSetBuilder.<ResolvedFile>stableOrder().addTransitive(children).add(root); |
| } else { |
| root = ResolvedFileFactory.directory(rootInfo.realPath); |
| } |
| return RecursiveFilesystemTraversalValue.of(root, paths.build()); |
| } |
| |
| private static HasDigest hashDirectorySymlink( |
| NestedSet<ResolvedFile> children, HasDigest metadata) { |
| // If the root is a directory symlink, the associated FileStateValue does not change when the |
| // linked directory's contents change, so we can't use the FileStateValue as metadata like we |
| // do with other ResolvedFile kinds. Instead we compute a metadata hash from the child |
| // elements and return that as the ResolvedFile's metadata hash. |
| Fingerprint fp = new Fingerprint(); |
| fp.addBytes(metadata.getDigest()); |
| for (ResolvedFile file : children.toList()) { |
| fp.addPath(file.getNameInSymlinkTree()); |
| fp.addBytes(file.getMetadata().getDigest()); |
| } |
| byte[] result = fp.digestAndReset(); |
| return new HasDigest.ByteStringDigest(result); |
| } |
| |
| private static SkyValue getDependentSkyValue(Environment env, SkyKey key) |
| throws MissingDepException, InterruptedException { |
| SkyValue value = env.getValue(key); |
| if (env.valuesMissing()) { |
| throw new MissingDepException(); |
| } |
| return value; |
| } |
| |
| /** |
| * Requests Skyframe to compute the dependent values and returns them. |
| * |
| * <p>The keys must all be {@link SkyFunctions#RECURSIVE_FILESYSTEM_TRAVERSAL} keys. |
| */ |
| private Collection<RecursiveFilesystemTraversalValue> traverseChildren( |
| Environment env, Iterable<SkyKey> keys, boolean inline) |
| throws MissingDepException, InterruptedException, |
| RecursiveFilesystemTraversalFunctionException { |
| Map<SkyKey, SkyValue> values; |
| if (inline) { |
| // Don't create Skyframe nodes for a recursive traversal over the output tree. |
| // Instead, inline the recursion in the top-level request. |
| values = new HashMap<>(); |
| for (SkyKey depKey : keys) { |
| values.put(depKey, compute(depKey, env)); |
| } |
| } else { |
| values = env.getValues(keys); |
| } |
| if (env.valuesMissing()) { |
| throw new MissingDepException(); |
| } |
| |
| return Collections2.transform(values.values(), RecursiveFilesystemTraversalValue.class::cast); |
| } |
| |
| } |