| // 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.concurrent.Uninterruptibles.callUninterruptibly; |
| import static com.google.devtools.build.lib.skyframe.ArtifactConflictFinder.ACTION_CONFLICTS; |
| import static com.google.devtools.build.lib.skyframe.ArtifactConflictFinder.NUM_JOBS; |
| |
| import com.google.common.annotations.VisibleForTesting; |
| import com.google.common.base.Preconditions; |
| import com.google.common.base.Predicate; |
| import com.google.common.base.Predicates; |
| import com.google.common.base.Stopwatch; |
| import com.google.common.base.Throwables; |
| import com.google.common.cache.Cache; |
| import com.google.common.cache.CacheBuilder; |
| import com.google.common.collect.ArrayListMultimap; |
| import com.google.common.collect.Collections2; |
| import com.google.common.collect.ImmutableCollection; |
| import com.google.common.collect.ImmutableList; |
| import com.google.common.collect.ImmutableMap; |
| import com.google.common.collect.ImmutableMultimap; |
| import com.google.common.collect.ImmutableSet; |
| import com.google.common.collect.ImmutableSortedSet; |
| import com.google.common.collect.Iterables; |
| import com.google.common.collect.Multimap; |
| import com.google.common.collect.Range; |
| import com.google.common.collect.Sets; |
| import com.google.common.eventbus.EventBus; |
| import com.google.common.flogger.GoogleLogger; |
| import com.google.common.hash.HashFunction; |
| import com.google.devtools.build.lib.actions.ActionAnalysisMetadata; |
| import com.google.devtools.build.lib.actions.ActionCacheChecker; |
| import com.google.devtools.build.lib.actions.ActionExecutionStatusReporter; |
| import com.google.devtools.build.lib.actions.ActionGraph; |
| import com.google.devtools.build.lib.actions.ActionInputMap; |
| import com.google.devtools.build.lib.actions.ActionInputPrefetcher; |
| import com.google.devtools.build.lib.actions.ActionKeyContext; |
| import com.google.devtools.build.lib.actions.ActionLogBufferPathGenerator; |
| import com.google.devtools.build.lib.actions.ActionLookupData; |
| import com.google.devtools.build.lib.actions.ActionLookupKey; |
| import com.google.devtools.build.lib.actions.ActionLookupValue; |
| import com.google.devtools.build.lib.actions.Actions; |
| import com.google.devtools.build.lib.actions.Artifact; |
| import com.google.devtools.build.lib.actions.ArtifactFactory; |
| import com.google.devtools.build.lib.actions.ArtifactPathResolver; |
| import com.google.devtools.build.lib.actions.ArtifactRoot; |
| import com.google.devtools.build.lib.actions.CommandLineExpansionException; |
| import com.google.devtools.build.lib.actions.CompletionContext.PathResolverFactory; |
| import com.google.devtools.build.lib.actions.EnvironmentalExecException; |
| import com.google.devtools.build.lib.actions.Executor; |
| import com.google.devtools.build.lib.actions.FileStateType; |
| import com.google.devtools.build.lib.actions.FileStateValue; |
| import com.google.devtools.build.lib.actions.FileValue; |
| import com.google.devtools.build.lib.actions.FilesetOutputSymlink; |
| import com.google.devtools.build.lib.actions.MetadataProvider; |
| import com.google.devtools.build.lib.actions.ResourceManager; |
| import com.google.devtools.build.lib.actions.UserExecException; |
| import com.google.devtools.build.lib.analysis.AnalysisProtos.ActionGraphContainer; |
| import com.google.devtools.build.lib.analysis.AspectCollection.AspectDeps; |
| import com.google.devtools.build.lib.analysis.AspectValue; |
| import com.google.devtools.build.lib.analysis.BlazeDirectories; |
| import com.google.devtools.build.lib.analysis.ConfigurationsCollector; |
| import com.google.devtools.build.lib.analysis.ConfigurationsResult; |
| import com.google.devtools.build.lib.analysis.ConfiguredAspect; |
| import com.google.devtools.build.lib.analysis.ConfiguredRuleClassProvider; |
| import com.google.devtools.build.lib.analysis.ConfiguredTarget; |
| import com.google.devtools.build.lib.analysis.ConfiguredTargetValue; |
| import com.google.devtools.build.lib.analysis.Dependency; |
| import com.google.devtools.build.lib.analysis.DependencyKey; |
| import com.google.devtools.build.lib.analysis.DuplicateException; |
| import com.google.devtools.build.lib.analysis.PlatformOptions; |
| import com.google.devtools.build.lib.analysis.TopLevelArtifactContext; |
| import com.google.devtools.build.lib.analysis.WorkspaceStatusAction; |
| import com.google.devtools.build.lib.analysis.WorkspaceStatusAction.Factory; |
| import com.google.devtools.build.lib.analysis.config.BuildConfiguration; |
| import com.google.devtools.build.lib.analysis.config.BuildConfigurationCollection; |
| import com.google.devtools.build.lib.analysis.config.BuildOptions; |
| import com.google.devtools.build.lib.analysis.config.BuildOptions.OptionsDiffForReconstruction; |
| import com.google.devtools.build.lib.analysis.config.BuildOptionsView; |
| import com.google.devtools.build.lib.analysis.config.ConfigurationResolver; |
| import com.google.devtools.build.lib.analysis.config.CoreOptions; |
| import com.google.devtools.build.lib.analysis.config.Fragment; |
| import com.google.devtools.build.lib.analysis.config.FragmentClassSet; |
| import com.google.devtools.build.lib.analysis.config.HostTransition; |
| import com.google.devtools.build.lib.analysis.config.InvalidConfigurationException; |
| import com.google.devtools.build.lib.analysis.config.transitions.ConfigurationTransition; |
| import com.google.devtools.build.lib.analysis.config.transitions.NoTransition; |
| import com.google.devtools.build.lib.analysis.config.transitions.NullTransition; |
| import com.google.devtools.build.lib.analysis.configuredtargets.InputFileConfiguredTarget; |
| import com.google.devtools.build.lib.analysis.configuredtargets.MergedConfiguredTarget; |
| import com.google.devtools.build.lib.analysis.configuredtargets.OutputFileConfiguredTarget; |
| import com.google.devtools.build.lib.analysis.starlark.StarlarkTransition; |
| import com.google.devtools.build.lib.analysis.starlark.StarlarkTransition.TransitionException; |
| import com.google.devtools.build.lib.bugreport.BugReport; |
| import com.google.devtools.build.lib.bugreport.BugReporter; |
| import com.google.devtools.build.lib.buildeventstream.BuildEventStreamProtos; |
| import com.google.devtools.build.lib.buildtool.BuildRequestOptions; |
| import com.google.devtools.build.lib.cmdline.Label; |
| import com.google.devtools.build.lib.cmdline.PackageIdentifier; |
| import com.google.devtools.build.lib.cmdline.TargetParsingException; |
| import com.google.devtools.build.lib.collect.compacthashset.CompactHashSet; |
| import com.google.devtools.build.lib.collect.nestedset.NestedSetExpander; |
| import com.google.devtools.build.lib.concurrent.NamedForkJoinPool; |
| import com.google.devtools.build.lib.concurrent.Sharder; |
| import com.google.devtools.build.lib.concurrent.ThreadSafety; |
| import com.google.devtools.build.lib.concurrent.ThreadSafety.ThreadCompatible; |
| import com.google.devtools.build.lib.events.ErrorSensingEventHandler; |
| import com.google.devtools.build.lib.events.Event; |
| import com.google.devtools.build.lib.events.EventHandler; |
| import com.google.devtools.build.lib.events.ExtendedEventHandler; |
| import com.google.devtools.build.lib.events.Reporter; |
| import com.google.devtools.build.lib.io.FileSymlinkCycleUniquenessFunction; |
| import com.google.devtools.build.lib.io.FileSymlinkInfiniteExpansionUniquenessFunction; |
| import com.google.devtools.build.lib.packages.BuildFileContainsErrorsException; |
| import com.google.devtools.build.lib.packages.BuildFileName; |
| import com.google.devtools.build.lib.packages.NoSuchPackageException; |
| import com.google.devtools.build.lib.packages.NoSuchTargetException; |
| import com.google.devtools.build.lib.packages.NoSuchThingException; |
| import com.google.devtools.build.lib.packages.Package; |
| import com.google.devtools.build.lib.packages.Package.Builder.PackageSettings; |
| import com.google.devtools.build.lib.packages.Package.ConfigSettingVisibilityPolicy; |
| import com.google.devtools.build.lib.packages.PackageFactory; |
| import com.google.devtools.build.lib.packages.RuleClassProvider; |
| import com.google.devtools.build.lib.packages.RuleVisibility; |
| import com.google.devtools.build.lib.packages.WorkspaceFileValue; |
| import com.google.devtools.build.lib.packages.semantics.BuildLanguageOptions; |
| import com.google.devtools.build.lib.pkgcache.LoadingOptions; |
| import com.google.devtools.build.lib.pkgcache.PackageManager; |
| import com.google.devtools.build.lib.pkgcache.PackageOptions; |
| import com.google.devtools.build.lib.pkgcache.PathPackageLocator; |
| import com.google.devtools.build.lib.pkgcache.QueryTransitivePackagePreloader; |
| import com.google.devtools.build.lib.pkgcache.TargetParsingPhaseTimeEvent; |
| import com.google.devtools.build.lib.pkgcache.TargetPatternPreloader; |
| import com.google.devtools.build.lib.pkgcache.TestFilter; |
| import com.google.devtools.build.lib.profiler.AutoProfiler; |
| import com.google.devtools.build.lib.profiler.GoogleAutoProfilerUtils; |
| import com.google.devtools.build.lib.profiler.Profiler; |
| import com.google.devtools.build.lib.profiler.SilentCloseable; |
| import com.google.devtools.build.lib.query2.common.UniverseScope; |
| import com.google.devtools.build.lib.remote.options.RemoteOptions; |
| import com.google.devtools.build.lib.remote.options.RemoteOutputsMode; |
| import com.google.devtools.build.lib.repository.ExternalPackageHelper; |
| import com.google.devtools.build.lib.rules.repository.ResolvedFileFunction; |
| import com.google.devtools.build.lib.rules.repository.ResolvedHashesFunction; |
| import com.google.devtools.build.lib.runtime.KeepGoingOption; |
| import com.google.devtools.build.lib.server.FailureDetails; |
| import com.google.devtools.build.lib.server.FailureDetails.BuildConfiguration.Code; |
| import com.google.devtools.build.lib.server.FailureDetails.FailureDetail; |
| import com.google.devtools.build.lib.server.FailureDetails.TargetPatterns; |
| import com.google.devtools.build.lib.skyframe.ArtifactConflictFinder.ConflictException; |
| import com.google.devtools.build.lib.skyframe.AspectValueKey.AspectKey; |
| import com.google.devtools.build.lib.skyframe.DirtinessCheckerUtils.FileDirtinessChecker; |
| import com.google.devtools.build.lib.skyframe.ExternalFilesHelper.ExternalFileAction; |
| import com.google.devtools.build.lib.skyframe.FileFunction.NonexistentFileReceiver; |
| import com.google.devtools.build.lib.skyframe.MetadataConsumerForMetrics.FilesMetricConsumer; |
| import com.google.devtools.build.lib.skyframe.PackageFunction.ActionOnIOExceptionReadingBuildFile; |
| import com.google.devtools.build.lib.skyframe.PackageFunction.IncrementalityIntent; |
| import com.google.devtools.build.lib.skyframe.PackageFunction.LoadedPackageCacheEntry; |
| import com.google.devtools.build.lib.skyframe.PackageLookupFunction.CrossRepositoryLabelViolationStrategy; |
| import com.google.devtools.build.lib.skyframe.SkyframeActionExecutor.ActionCompletedReceiver; |
| import com.google.devtools.build.lib.skyframe.SkyframeActionExecutor.ProgressSupplier; |
| import com.google.devtools.build.lib.skyframe.trimming.TrimmedConfigurationCache; |
| import com.google.devtools.build.lib.util.AbruptExitException; |
| import com.google.devtools.build.lib.util.DetailedExitCode; |
| import com.google.devtools.build.lib.util.ResourceUsage; |
| import com.google.devtools.build.lib.util.TestType; |
| import com.google.devtools.build.lib.util.io.TimestampGranularityMonitor; |
| import com.google.devtools.build.lib.vfs.Dirent; |
| import com.google.devtools.build.lib.vfs.FileSystem; |
| import com.google.devtools.build.lib.vfs.ModifiedFileSet; |
| import com.google.devtools.build.lib.vfs.OutputService; |
| 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.UnixGlob; |
| import com.google.devtools.build.skyframe.BuildDriver; |
| import com.google.devtools.build.skyframe.CycleInfo; |
| import com.google.devtools.build.skyframe.CyclesReporter; |
| import com.google.devtools.build.skyframe.Differencer; |
| import com.google.devtools.build.skyframe.Differencer.DiffWithDelta.Delta; |
| import com.google.devtools.build.skyframe.ErrorInfo; |
| import com.google.devtools.build.skyframe.EvaluationContext; |
| import com.google.devtools.build.skyframe.EvaluationProgressReceiver; |
| import com.google.devtools.build.skyframe.EvaluationResult; |
| import com.google.devtools.build.skyframe.EventFilter; |
| import com.google.devtools.build.skyframe.GraphInconsistencyReceiver; |
| import com.google.devtools.build.skyframe.ImmutableDiff; |
| import com.google.devtools.build.skyframe.InMemoryMemoizingEvaluator; |
| import com.google.devtools.build.skyframe.Injectable; |
| import com.google.devtools.build.skyframe.MemoizingEvaluator; |
| import com.google.devtools.build.skyframe.MemoizingEvaluator.EvaluatorSupplier; |
| import com.google.devtools.build.skyframe.NodeEntry; |
| import com.google.devtools.build.skyframe.SkyFunction; |
| import com.google.devtools.build.skyframe.SkyFunctionName; |
| import com.google.devtools.build.skyframe.SkyKey; |
| import com.google.devtools.build.skyframe.SkyValue; |
| import com.google.devtools.build.skyframe.WalkableGraph; |
| import com.google.devtools.build.skyframe.WalkableGraph.WalkableGraphFactory; |
| import com.google.devtools.common.options.OptionsParsingException; |
| import com.google.devtools.common.options.OptionsProvider; |
| import com.google.errorprone.annotations.ForOverride; |
| import java.io.IOException; |
| import java.io.PrintStream; |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.HashSet; |
| import java.util.Iterator; |
| import java.util.LinkedHashMap; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Optional; |
| import java.util.Set; |
| import java.util.UUID; |
| import java.util.concurrent.Callable; |
| import java.util.concurrent.Semaphore; |
| import java.util.concurrent.atomic.AtomicBoolean; |
| import java.util.concurrent.atomic.AtomicInteger; |
| import java.util.concurrent.atomic.AtomicReference; |
| import java.util.function.Consumer; |
| import java.util.function.Supplier; |
| import java.util.stream.Collectors; |
| import javax.annotation.Nullable; |
| import net.starlark.java.eval.StarlarkSemantics; |
| |
| /** |
| * A helper object to support Skyframe-driven execution. |
| * |
| * <p>This object is mostly used to inject external state, such as the executor engine or some |
| * additional artifacts (workspace status and build info artifacts) into SkyFunctions for use during |
| * the build. |
| */ |
| public abstract class SkyframeExecutor implements WalkableGraphFactory, ConfigurationsCollector { |
| private static final GoogleLogger logger = GoogleLogger.forEnclosingClass(); |
| |
| // We delete any value that can hold an action -- all subclasses of ActionLookupKey. |
| // Also remove ArtifactNestedSetValues to prevent memory leak (b/143940221). |
| protected static final Predicate<SkyKey> ANALYSIS_INVALIDATING_PREDICATE = |
| k -> (k instanceof ArtifactNestedSetKey || k instanceof ActionLookupKey); |
| |
| private final EvaluatorSupplier evaluatorSupplier; |
| protected MemoizingEvaluator memoizingEvaluator; |
| private final MemoizingEvaluator.EmittedEventState emittedEventState = |
| new MemoizingEvaluator.EmittedEventState(); |
| private final PackageFactory pkgFactory; |
| private final WorkspaceStatusAction.Factory workspaceStatusActionFactory; |
| private final FileSystem fileSystem; |
| protected final BlazeDirectories directories; |
| protected final ExternalFilesHelper externalFilesHelper; |
| private final GraphInconsistencyReceiver graphInconsistencyReceiver; |
| private final BugReporter bugReporter; |
| |
| /** |
| * Measures source artifacts read this build. Does not include cached artifacts, so is less useful |
| * on incremental builds. |
| */ |
| private final FilesMetricConsumer sourceArtifactsSeen = new FilesMetricConsumer(); |
| |
| private final FilesMetricConsumer outputArtifactsSeen = new FilesMetricConsumer(); |
| private final FilesMetricConsumer outputArtifactsFromActionCache = new FilesMetricConsumer(); |
| private final FilesMetricConsumer topLevelArtifactsMetric = new FilesMetricConsumer(); |
| |
| @Nullable protected OutputService outputService; |
| |
| // TODO(bazel-team): Figure out how to handle value builders that block internally. Blocking |
| // operations may need to be handled in another (bigger?) thread pool. Also, we should detect |
| // the number of cores and use that as the thread-pool size for CPU-bound operations. |
| // I just bumped this to 200 to get reasonable execution phase performance; that may cause |
| // significant overhead for CPU-bound processes (i.e. analysis). [skyframe-analysis] |
| public static final int DEFAULT_THREAD_COUNT = |
| // Reduce thread count while running tests of Bazel. Test cases are typically small, and large |
| // thread pools vying for a relatively small number of CPU cores may induce non-optimal |
| // performance. |
| TestType.isInTest() ? 5 : 200; |
| |
| // The limit of how many times we will traverse through an exception chain when catching a |
| // target parsing exception. |
| private static final int EXCEPTION_TRAVERSAL_LIMIT = 10; |
| |
| // Cache of partially constructed Package instances, stored between reruns of the PackageFunction |
| // (because of missing dependencies, within the same evaluate() run) to avoid loading the same |
| // package twice (first time loading to find load()ed bzl files and declare Skyframe |
| // dependencies). |
| private final Cache<PackageIdentifier, LoadedPackageCacheEntry> packageFunctionCache = |
| newPkgFunctionCache(); |
| // Cache of parsed BUILD files, for PackageFunction. Same motivation as above. |
| private final Cache<PackageIdentifier, PackageFunction.CompiledBuildFile> compiledBuildFileCache = |
| newCompiledBuildFileCache(); |
| |
| // Cache of parsed bzl files, for use when we're inlining BzlCompileFunction in |
| // BzlLoadFunction. See the comments in BzlLoadFunction for motivations and details. |
| private final Cache<BzlCompileValue.Key, BzlCompileValue> bzlCompileCache = |
| CacheBuilder.newBuilder().build(); |
| |
| private final AtomicInteger numPackagesLoaded = new AtomicInteger(0); |
| @Nullable private final PackageProgressReceiver packageProgress; |
| @Nullable private final ConfiguredTargetProgressReceiver configuredTargetProgress; |
| |
| private final SkyframeBuildView skyframeBuildView; |
| private ActionLogBufferPathGenerator actionLogBufferPathGenerator; |
| |
| private BuildDriver buildDriver; |
| |
| private final Consumer<SkyframeExecutor> skyframeExecutorConsumerOnInit; |
| |
| // AtomicReferences are used here as mutable boxes shared with value builders. |
| private final AtomicBoolean showLoadingProgress = new AtomicBoolean(); |
| protected final AtomicReference<UnixGlob.FilesystemCalls> syscalls = |
| new AtomicReference<>(UnixGlob.DEFAULT_SYSCALLS); |
| protected final AtomicReference<PathPackageLocator> pkgLocator = new AtomicReference<>(); |
| protected final AtomicReference<ImmutableSet<PackageIdentifier>> deletedPackages = |
| new AtomicReference<>(ImmutableSet.<PackageIdentifier>of()); |
| private final AtomicReference<EventBus> eventBus = new AtomicReference<>(); |
| protected final AtomicReference<TimestampGranularityMonitor> tsgm = new AtomicReference<>(); |
| protected final AtomicReference<Map<String, String>> clientEnv = new AtomicReference<>(); |
| |
| private final ArtifactFactory artifactFactory; |
| private final ActionKeyContext actionKeyContext; |
| |
| protected boolean active = true; |
| private final SkyframePackageManager packageManager; |
| |
| /** Used to lock evaluator on legacy calls to get existing values. */ |
| private final Object valueLookupLock = new Object(); |
| |
| private final AtomicReference<ActionExecutionStatusReporter> statusReporterRef = |
| new AtomicReference<>(); |
| protected final SkyframeActionExecutor skyframeActionExecutor; |
| private ActionExecutionFunction actionExecutionFunction; |
| protected SkyframeProgressReceiver progressReceiver; |
| private CyclesReporter cyclesReporter = null; |
| |
| @VisibleForTesting boolean lastAnalysisDiscarded = false; |
| |
| private boolean analysisCacheDiscarded = false; |
| |
| private final ImmutableMap<SkyFunctionName, SkyFunction> extraSkyFunctions; |
| |
| protected SkyframeIncrementalBuildMonitor incrementalBuildMonitor = |
| new SkyframeIncrementalBuildMonitor(); |
| |
| private final SkyFunction ignoredPackagePrefixesFunction; |
| |
| private final ConfiguredRuleClassProvider ruleClassProvider; |
| |
| private final CrossRepositoryLabelViolationStrategy crossRepositoryLabelViolationStrategy; |
| |
| private final ImmutableList<BuildFileName> buildFilesByPriority; |
| |
| private final ExternalPackageHelper externalPackageHelper; |
| |
| private final ActionOnIOExceptionReadingBuildFile actionOnIOExceptionReadingBuildFile; |
| |
| private final boolean shouldUnblockCpuWorkWhenFetchingDeps; |
| |
| private final BuildOptions defaultBuildOptions; |
| |
| private PerBuildSyscallCache perBuildSyscallCache; |
| private int lastConcurrencyLevel = -1; |
| |
| private final PathResolverFactory pathResolverFactory = new PathResolverFactoryImpl(); |
| @Nullable private final NonexistentFileReceiver nonexistentFileReceiver; |
| |
| private final TrimmedConfigurationCache<SkyKey, Label, OptionsDiffForReconstruction> |
| trimmingCache = TrimmedConfigurationProgressReceiver.buildCache(); |
| private final TrimmedConfigurationProgressReceiver trimmingListener = |
| new TrimmedConfigurationProgressReceiver(trimmingCache); |
| |
| private boolean siblingRepositoryLayout = false; |
| |
| private Map<String, String> lastRemoteDefaultExecProperties; |
| private RemoteOutputsMode lastRemoteOutputsMode; |
| |
| class PathResolverFactoryImpl implements PathResolverFactory { |
| @Override |
| public boolean shouldCreatePathResolverForArtifactValues() { |
| return outputService != null && outputService.supportsPathResolverForArtifactValues(); |
| } |
| |
| @Override |
| public ArtifactPathResolver createPathResolverForArtifactValues( |
| ActionInputMap actionInputMap, |
| Map<Artifact, ImmutableCollection<Artifact>> expandedArtifacts, |
| Map<Artifact, ImmutableList<FilesetOutputSymlink>> filesets, |
| String workspaceName) |
| throws IOException { |
| Preconditions.checkState(shouldCreatePathResolverForArtifactValues()); |
| return outputService.createPathResolverForArtifactValues( |
| directories.getExecRoot(workspaceName).asFragment(), |
| directories.getRelativeOutputPath(), |
| fileSystem, |
| getPathEntries(), |
| actionInputMap, |
| expandedArtifacts, |
| filesets); |
| } |
| } |
| |
| protected SkyframeExecutor( |
| Consumer<SkyframeExecutor> skyframeExecutorConsumerOnInit, |
| EvaluatorSupplier evaluatorSupplier, |
| PackageFactory pkgFactory, |
| FileSystem fileSystem, |
| BlazeDirectories directories, |
| ActionKeyContext actionKeyContext, |
| Factory workspaceStatusActionFactory, |
| ImmutableMap<SkyFunctionName, SkyFunction> extraSkyFunctions, |
| ExternalFileAction externalFileAction, |
| SkyFunction ignoredPackagePrefixesFunction, |
| CrossRepositoryLabelViolationStrategy crossRepositoryLabelViolationStrategy, |
| ImmutableList<BuildFileName> buildFilesByPriority, |
| ExternalPackageHelper externalPackageHelper, |
| ActionOnIOExceptionReadingBuildFile actionOnIOExceptionReadingBuildFile, |
| boolean shouldUnblockCpuWorkWhenFetchingDeps, |
| GraphInconsistencyReceiver graphInconsistencyReceiver, |
| BuildOptions defaultBuildOptions, |
| @Nullable PackageProgressReceiver packageProgress, |
| @Nullable ConfiguredTargetProgressReceiver configuredTargetProgress, |
| @Nullable NonexistentFileReceiver nonexistentFileReceiver, |
| @Nullable ManagedDirectoriesKnowledge managedDirectoriesKnowledge, |
| BugReporter bugReporter) { |
| // Strictly speaking, these arguments are not required for initialization, but all current |
| // callsites have them at hand, so we might as well set them during construction. |
| this.skyframeExecutorConsumerOnInit = skyframeExecutorConsumerOnInit; |
| this.evaluatorSupplier = evaluatorSupplier; |
| this.pkgFactory = pkgFactory; |
| this.shouldUnblockCpuWorkWhenFetchingDeps = shouldUnblockCpuWorkWhenFetchingDeps; |
| this.graphInconsistencyReceiver = graphInconsistencyReceiver; |
| this.nonexistentFileReceiver = nonexistentFileReceiver; |
| this.bugReporter = bugReporter; |
| this.pkgFactory.setSyscalls(syscalls); |
| this.workspaceStatusActionFactory = workspaceStatusActionFactory; |
| this.packageManager = |
| new SkyframePackageManager( |
| new SkyframePackageLoader(), |
| new QueryTransitivePackagePreloader(this::getDriver), |
| syscalls, |
| pkgLocator, |
| numPackagesLoaded, |
| this); |
| this.fileSystem = fileSystem; |
| this.directories = Preconditions.checkNotNull(directories); |
| this.actionKeyContext = Preconditions.checkNotNull(actionKeyContext); |
| this.ignoredPackagePrefixesFunction = ignoredPackagePrefixesFunction; |
| this.extraSkyFunctions = extraSkyFunctions; |
| |
| this.ruleClassProvider = (ConfiguredRuleClassProvider) pkgFactory.getRuleClassProvider(); |
| this.defaultBuildOptions = defaultBuildOptions; |
| this.skyframeActionExecutor = |
| new SkyframeActionExecutor( |
| actionKeyContext, |
| outputArtifactsSeen, |
| outputArtifactsFromActionCache, |
| statusReporterRef, |
| this::getPathEntries, |
| PathFragment.create(directories.getRelativeOutputPath())); |
| this.artifactFactory = |
| new ArtifactFactory( |
| /* execRootParent= */ directories.getExecRootBase(), |
| directories.getRelativeOutputPath()); |
| this.skyframeBuildView = |
| new SkyframeBuildView(artifactFactory, this, ruleClassProvider, actionKeyContext); |
| this.externalFilesHelper = |
| ExternalFilesHelper.create( |
| pkgLocator, |
| externalFileAction, |
| directories, |
| managedDirectoriesKnowledge != null |
| ? managedDirectoriesKnowledge |
| : ManagedDirectoriesKnowledge.NO_MANAGED_DIRECTORIES, |
| externalPackageHelper); |
| this.crossRepositoryLabelViolationStrategy = crossRepositoryLabelViolationStrategy; |
| this.buildFilesByPriority = buildFilesByPriority; |
| this.externalPackageHelper = externalPackageHelper; |
| this.actionOnIOExceptionReadingBuildFile = actionOnIOExceptionReadingBuildFile; |
| this.packageProgress = packageProgress; |
| this.configuredTargetProgress = configuredTargetProgress; |
| } |
| |
| private ImmutableMap<SkyFunctionName, SkyFunction> skyFunctions(PackageFactory pkgFactory) { |
| ConfiguredRuleClassProvider ruleClassProvider = |
| (ConfiguredRuleClassProvider) pkgFactory.getRuleClassProvider(); |
| BzlLoadFunction bzlLoadFunctionForInliningPackageAndWorkspaceNodes = |
| getBzlLoadFunctionForInliningPackageAndWorkspaceNodes(); |
| // TODO(janakr): use this semaphore to bound memory usage for SkyFunctions besides |
| // ConfiguredTargetFunction that may have a large temporary memory blow-up. |
| Semaphore cpuBoundSemaphore = new Semaphore(ResourceUsage.getAvailableProcessors()); |
| |
| // We don't check for duplicates in order to allow extraSkyfunctions to override existing |
| // entries. |
| Map<SkyFunctionName, SkyFunction> map = new HashMap<>(); |
| // IF YOU ADD A NEW SKYFUNCTION: If your Skyfunction can be used transitively by package |
| // loading, make sure to register it in AbstractPackageLoader as well. |
| map.put(SkyFunctions.PRECOMPUTED, new PrecomputedFunction()); |
| map.put(SkyFunctions.CLIENT_ENVIRONMENT_VARIABLE, new ClientEnvironmentFunction(clientEnv)); |
| map.put(SkyFunctions.ACTION_ENVIRONMENT_VARIABLE, new ActionEnvironmentFunction()); |
| map.put(FileStateValue.FILE_STATE, newFileStateFunction()); |
| map.put(SkyFunctions.DIRECTORY_LISTING_STATE, newDirectoryListingStateFunction()); |
| map.put(FileSymlinkCycleUniquenessFunction.NAME, new FileSymlinkCycleUniquenessFunction()); |
| map.put( |
| FileSymlinkInfiniteExpansionUniquenessFunction.NAME, |
| new FileSymlinkInfiniteExpansionUniquenessFunction()); |
| map.put(FileValue.FILE, new FileFunction(pkgLocator, nonexistentFileReceiver)); |
| map.put(SkyFunctions.DIRECTORY_LISTING, new DirectoryListingFunction()); |
| map.put( |
| SkyFunctions.PACKAGE_LOOKUP, |
| new PackageLookupFunction( |
| deletedPackages, |
| crossRepositoryLabelViolationStrategy, |
| buildFilesByPriority, |
| externalPackageHelper)); |
| map.put(SkyFunctions.CONTAINING_PACKAGE_LOOKUP, new ContainingPackageLookupFunction()); |
| map.put( |
| SkyFunctions.BZL_COMPILE, // TODO rename |
| new BzlCompileFunction(pkgFactory, getHashFunction())); |
| map.put(SkyFunctions.STARLARK_BUILTINS, new StarlarkBuiltinsFunction(pkgFactory)); |
| map.put(SkyFunctions.BZL_LOAD, newBzlLoadFunction(ruleClassProvider, pkgFactory)); |
| map.put(SkyFunctions.GLOB, newGlobFunction()); |
| map.put(SkyFunctions.TARGET_PATTERN, new TargetPatternFunction()); |
| map.put(SkyFunctions.PREPARE_DEPS_OF_PATTERNS, new PrepareDepsOfPatternsFunction()); |
| map.put( |
| SkyFunctions.PREPARE_DEPS_OF_PATTERN, |
| new PrepareDepsOfPatternFunction(pkgLocator, traverseTestSuites())); |
| map.put( |
| SkyFunctions.PREPARE_TEST_SUITES_UNDER_DIRECTORY, |
| new PrepareTestSuitesUnderDirectoryFunction(directories)); |
| map.put( |
| SkyFunctions.PREPARE_DEPS_OF_TARGETS_UNDER_DIRECTORY, |
| new PrepareDepsOfTargetsUnderDirectoryFunction(directories)); |
| map.put(SkyFunctions.COLLECT_TARGETS_IN_PACKAGE, new CollectTargetsInPackageFunction()); |
| map.put(SkyFunctions.COLLECT_TEST_SUITES_IN_PACKAGE, new CollectTestSuitesInPackageFunction()); |
| map.put( |
| SkyFunctions.COLLECT_PACKAGES_UNDER_DIRECTORY, |
| newCollectPackagesUnderDirectoryFunction(directories)); |
| map.put(SkyFunctions.IGNORED_PACKAGE_PREFIXES, ignoredPackagePrefixesFunction); |
| map.put(SkyFunctions.TESTS_IN_SUITE, new TestExpansionFunction()); |
| map.put(SkyFunctions.TEST_SUITE_EXPANSION, new TestsForTargetPatternFunction()); |
| map.put( |
| SkyFunctions.TARGET_PATTERN_PHASE, new TargetPatternPhaseFunction(externalPackageHelper)); |
| map.put( |
| SkyFunctions.PREPARE_ANALYSIS_PHASE, |
| new PrepareAnalysisPhaseFunction(ruleClassProvider, defaultBuildOptions)); |
| map.put(SkyFunctions.RECURSIVE_PKG, new RecursivePkgFunction(directories)); |
| map.put( |
| SkyFunctions.PACKAGE, |
| new PackageFunction( |
| pkgFactory, |
| packageManager, |
| showLoadingProgress, |
| packageFunctionCache, |
| compiledBuildFileCache, |
| numPackagesLoaded, |
| bzlLoadFunctionForInliningPackageAndWorkspaceNodes, |
| packageProgress, |
| actionOnIOExceptionReadingBuildFile, |
| tracksStateForIncrementality() |
| ? IncrementalityIntent.INCREMENTAL |
| : IncrementalityIntent.NON_INCREMENTAL)); |
| map.put(SkyFunctions.PACKAGE_ERROR, new PackageErrorFunction()); |
| map.put(SkyFunctions.PACKAGE_ERROR_MESSAGE, new PackageErrorMessageFunction()); |
| map.put(SkyFunctions.TARGET_PATTERN_ERROR, new TargetPatternErrorFunction()); |
| map.put(TransitiveTargetKey.NAME, new TransitiveTargetFunction(ruleClassProvider)); |
| map.put(Label.TRANSITIVE_TRAVERSAL, getTransitiveTraversalFunction()); |
| map.put( |
| SkyFunctions.CONFIGURED_TARGET, |
| new ConfiguredTargetFunction( |
| new BuildViewProvider(), |
| ruleClassProvider, |
| cpuBoundSemaphore, |
| shouldStoreTransitivePackagesInLoadingAndAnalysis(), |
| shouldUnblockCpuWorkWhenFetchingDeps, |
| defaultBuildOptions, |
| configuredTargetProgress)); |
| map.put( |
| SkyFunctions.ASPECT, |
| new AspectFunction( |
| new BuildViewProvider(), |
| ruleClassProvider, |
| shouldStoreTransitivePackagesInLoadingAndAnalysis(), |
| defaultBuildOptions)); |
| map.put(SkyFunctions.LOAD_STARLARK_ASPECT, new ToplevelStarlarkAspectFunction()); |
| map.put(SkyFunctions.ACTION_LOOKUP_CONFLICT_FINDING, new ActionLookupConflictFindingFunction()); |
| map.put( |
| SkyFunctions.TOP_LEVEL_ACTION_LOOKUP_CONFLICT_FINDING, |
| new TopLevelActionLookupConflictFindingFunction()); |
| map.put( |
| SkyFunctions.BUILD_CONFIGURATION, |
| new BuildConfigurationFunction(directories, ruleClassProvider, defaultBuildOptions)); |
| map.put(SkyFunctions.WORKSPACE_NAME, new WorkspaceNameFunction()); |
| map.put( |
| WorkspaceFileValue.WORKSPACE_FILE, |
| new WorkspaceFileFunction( |
| ruleClassProvider, |
| pkgFactory, |
| directories, |
| bzlLoadFunctionForInliningPackageAndWorkspaceNodes)); |
| map.put(SkyFunctions.EXTERNAL_PACKAGE, new ExternalPackageFunction(externalPackageHelper)); |
| map.put( |
| SkyFunctions.TARGET_COMPLETION, |
| TargetCompletor.targetCompletionFunction( |
| pathResolverFactory, skyframeActionExecutor, topLevelArtifactsMetric, bugReporter)); |
| map.put( |
| SkyFunctions.ASPECT_COMPLETION, |
| AspectCompletor.aspectCompletionFunction( |
| pathResolverFactory, skyframeActionExecutor, topLevelArtifactsMetric, bugReporter)); |
| map.put(SkyFunctions.TEST_COMPLETION, new TestCompletionFunction()); |
| map.put( |
| Artifact.ARTIFACT, |
| new ArtifactFunction( |
| () -> !skyframeActionExecutor.actionFileSystemType().inMemoryFileSystem(), |
| sourceArtifactsSeen)); |
| map.put( |
| SkyFunctions.BUILD_INFO_COLLECTION, |
| new BuildInfoCollectionFunction(actionKeyContext, artifactFactory)); |
| map.put(SkyFunctions.BUILD_INFO, new WorkspaceStatusFunction(this::makeWorkspaceStatusAction)); |
| map.put(SkyFunctions.COVERAGE_REPORT, new CoverageReportFunction(actionKeyContext)); |
| ActionExecutionFunction actionExecutionFunction = |
| new ActionExecutionFunction(skyframeActionExecutor, directories, tsgm, bugReporter); |
| map.put(SkyFunctions.ACTION_EXECUTION, actionExecutionFunction); |
| this.actionExecutionFunction = actionExecutionFunction; |
| map.put(SkyFunctions.ACTION_SKETCH, new ActionSketchFunction(actionKeyContext)); |
| map.put( |
| SkyFunctions.RECURSIVE_FILESYSTEM_TRAVERSAL, new RecursiveFilesystemTraversalFunction()); |
| map.put(SkyFunctions.FILESET_ENTRY, new FilesetEntryFunction(directories::getExecRoot)); |
| map.put( |
| SkyFunctions.ACTION_TEMPLATE_EXPANSION, |
| new ActionTemplateExpansionFunction(actionKeyContext)); |
| map.put( |
| SkyFunctions.LOCAL_REPOSITORY_LOOKUP, |
| new LocalRepositoryLookupFunction(externalPackageHelper)); |
| map.put( |
| SkyFunctions.REGISTERED_EXECUTION_PLATFORMS, new RegisteredExecutionPlatformsFunction()); |
| map.put(SkyFunctions.REGISTERED_TOOLCHAINS, new RegisteredToolchainsFunction()); |
| map.put(SkyFunctions.SINGLE_TOOLCHAIN_RESOLUTION, new SingleToolchainResolutionFunction()); |
| map.put(SkyFunctions.TOOLCHAIN_RESOLUTION, new ToolchainResolutionFunction()); |
| map.put(SkyFunctions.REPOSITORY_MAPPING, new RepositoryMappingFunction()); |
| map.put(SkyFunctions.RESOLVED_HASH_VALUES, new ResolvedHashesFunction()); |
| map.put(SkyFunctions.RESOLVED_FILE, new ResolvedFileFunction()); |
| map.put(SkyFunctions.PLATFORM_MAPPING, new PlatformMappingFunction()); |
| map.put(SkyFunctions.ARTIFACT_NESTED_SET, ArtifactNestedSetFunction.createInstance()); |
| map.putAll(extraSkyFunctions); |
| return ImmutableMap.copyOf(map); |
| } |
| |
| protected SkyFunction getTransitiveTraversalFunction() { |
| return new TransitiveTraversalFunction(); |
| } |
| |
| protected boolean traverseTestSuites() { |
| return false; |
| } |
| |
| protected SkyFunction newFileStateFunction() { |
| return new FileStateFunction(tsgm, syscalls, externalFilesHelper); |
| } |
| |
| protected SkyFunction newDirectoryListingStateFunction() { |
| return new DirectoryListingStateFunction(externalFilesHelper, syscalls); |
| } |
| |
| protected SkyFunction newGlobFunction() { |
| return new GlobFunction(/*alwaysUseDirListing=*/ false); |
| } |
| |
| protected SkyFunction newCollectPackagesUnderDirectoryFunction(BlazeDirectories directories) { |
| return new CollectPackagesUnderDirectoryFunction(directories); |
| } |
| |
| @Nullable |
| protected BzlLoadFunction getBzlLoadFunctionForInliningPackageAndWorkspaceNodes() { |
| return null; |
| } |
| |
| protected SkyFunction newBzlLoadFunction( |
| RuleClassProvider ruleClassProvider, PackageFactory pkgFactory) { |
| return BzlLoadFunction.create(this.pkgFactory, directories, getHashFunction(), bzlCompileCache); |
| } |
| |
| protected PerBuildSyscallCache newPerBuildSyscallCache(int concurrencyLevel) { |
| return PerBuildSyscallCache.newBuilder().setConcurrencyLevel(concurrencyLevel).build(); |
| } |
| |
| /** |
| * Gets a (possibly cached) syscalls cache, re-initialized each build. |
| * |
| * <p>We cache the syscalls cache if possible because construction of the cache is surprisingly |
| * expensive, and is on the critical path of null builds. |
| */ |
| protected final PerBuildSyscallCache getPerBuildSyscallCache(int concurrencyLevel) { |
| if (perBuildSyscallCache != null && lastConcurrencyLevel == concurrencyLevel) { |
| perBuildSyscallCache.clear(); |
| return perBuildSyscallCache; |
| } |
| lastConcurrencyLevel = concurrencyLevel; |
| perBuildSyscallCache = newPerBuildSyscallCache(concurrencyLevel); |
| return perBuildSyscallCache; |
| } |
| |
| public AtomicReference<UnixGlob.FilesystemCalls> getSyscalls() { |
| return syscalls; |
| } |
| |
| @ThreadCompatible |
| public void setActive(boolean active) { |
| this.active = active; |
| } |
| |
| protected void checkActive() { |
| Preconditions.checkState(active); |
| } |
| |
| public void configureActionExecutor( |
| MetadataProvider fileCache, ActionInputPrefetcher actionInputPrefetcher) { |
| skyframeActionExecutor.configure(fileCache, actionInputPrefetcher, NestedSetExpander.DEFAULT); |
| } |
| |
| public void dump(boolean summarize, PrintStream out) { |
| memoizingEvaluator.dump(summarize, out); |
| } |
| |
| public abstract void dumpPackages(PrintStream out); |
| |
| public void setOutputService(OutputService outputService) { |
| this.outputService = outputService; |
| } |
| |
| /** Inform this SkyframeExecutor that a new command is starting. */ |
| public void noteCommandStart() {} |
| |
| /** Notify listeners about changed files, and release any associated memory afterwards. */ |
| public void drainChangedFiles() { |
| incrementalBuildMonitor.alertListeners(getEventBus()); |
| incrementalBuildMonitor = null; |
| } |
| |
| public final BuildDriver getDriver() { |
| return buildDriver; |
| } |
| |
| public boolean wasAnalysisCacheDiscardedAndResetBit() { |
| boolean tmp = analysisCacheDiscarded; |
| analysisCacheDiscarded = false; |
| return tmp; |
| } |
| |
| /** |
| * This method exists only to allow a module to make a top-level Skyframe call during the |
| * transition to making it fully Skyframe-compatible. Do not add additional callers! |
| */ |
| public SkyValue evaluateSkyKeyForExecutionSetup( |
| final ExtendedEventHandler eventHandler, final SkyKey key) |
| throws EnvironmentalExecException, InterruptedException { |
| synchronized (valueLookupLock) { |
| // We evaluate in keepGoing mode because in the case that the graph does not store its |
| // edges, nokeepGoing builds are not allowed, whereas keepGoing builds are always |
| // permitted. |
| EvaluationResult<?> result = |
| evaluate( |
| ImmutableList.of(key), true, ResourceUsage.getAvailableProcessors(), eventHandler); |
| if (!result.hasError()) { |
| return Preconditions.checkNotNull(result.get(key), "%s %s", result, key); |
| } |
| ErrorInfo errorInfo = Preconditions.checkNotNull(result.getError(key), "%s %s", key, result); |
| Throwables.propagateIfPossible(errorInfo.getException(), EnvironmentalExecException.class); |
| if (errorInfo.getException() != null) { |
| throw new IllegalStateException(errorInfo.getException()); |
| } |
| throw new IllegalStateException(errorInfo.toString()); |
| } |
| } |
| |
| public abstract ActionGraphContainer getActionGraphContainer( |
| List<String> actionGraphTargets, boolean includeActionCmdLine, boolean includeArtifacts) |
| throws CommandLineExpansionException; |
| |
| class BuildViewProvider { |
| /** Returns the current {@link SkyframeBuildView} instance. */ |
| SkyframeBuildView getSkyframeBuildView() { |
| return skyframeBuildView; |
| } |
| } |
| |
| /** |
| * Must be called before the {@link SkyframeExecutor} can be used (should only be called in |
| * factory methods and as an implementation detail of {@link #resetEvaluator}). |
| */ |
| protected void init() { |
| progressReceiver = newSkyframeProgressReceiver(); |
| ImmutableMap<SkyFunctionName, SkyFunction> skyFunctions = skyFunctions(pkgFactory); |
| memoizingEvaluator = |
| evaluatorSupplier.create( |
| skyFunctions, |
| evaluatorDiffer(), |
| progressReceiver, |
| graphInconsistencyReceiver, |
| DEFAULT_FILTER_WITH_ACTIONS, |
| emittedEventState, |
| tracksStateForIncrementality()); |
| buildDriver = createBuildDriver(); |
| skyframeExecutorConsumerOnInit.accept(this); |
| } |
| |
| /** |
| * Use the fact that analysis of a target must occur before execution of that target, and in a |
| * separate Skyframe evaluation, to avoid propagating events from configured target nodes (and |
| * more generally action lookup nodes) to action execution nodes. We take advantage of the fact |
| * that if a node depends on an action lookup node and is not itself an action lookup node, then |
| * it is an execution-phase node: the action lookup nodes are terminal in the analysis phase. |
| */ |
| private static final EventFilter DEFAULT_FILTER_WITH_ACTIONS = |
| new EventFilter() { |
| @Override |
| public boolean storeEventsAndPosts() { |
| return true; |
| } |
| |
| @Override |
| public boolean apply(Event input) { |
| // Use the filtering defined in the default filter: no info/progress messages. |
| return InMemoryMemoizingEvaluator.DEFAULT_STORED_EVENT_FILTER.apply(input); |
| } |
| |
| @Override |
| public Predicate<SkyKey> depEdgeFilterForEventsAndPosts(SkyKey primaryKey) { |
| if (primaryKey instanceof ActionLookupKey) { |
| return Predicates.alwaysTrue(); |
| } else { |
| return NO_ACTION_LOOKUP; |
| } |
| } |
| }; |
| |
| private static final Predicate<SkyKey> NO_ACTION_LOOKUP = |
| (key) -> !(key instanceof ActionLookupKey); |
| |
| protected SkyframeProgressReceiver newSkyframeProgressReceiver() { |
| return new SkyframeProgressReceiver(); |
| } |
| |
| /** Reinitializes the Skyframe evaluator, dropping all previously computed values. */ |
| public void resetEvaluator() { |
| init(); |
| emittedEventState.clear(); |
| clearTrimmingCache(); |
| skyframeBuildView.reset(); |
| } |
| |
| /** |
| * Notifies the executor that the command is complete. May safely be called multiple times for a |
| * single command, so callers should err on the side of calling it more frequently. Should be |
| * idempotent, so that calls after the first one in the same evaluation should be quick. |
| */ |
| public void notifyCommandComplete(ExtendedEventHandler eventHandler) throws InterruptedException { |
| memoizingEvaluator.noteEvaluationsAtSameVersionMayBeFinished(eventHandler); |
| } |
| |
| /** |
| * Notifies the executor to post logging stats when the server is crashing, so that logging is |
| * still available even when the server crashes. |
| */ |
| public void postLoggingStatsWhenCrashing(ExtendedEventHandler eventHandler) { |
| memoizingEvaluator.postLoggingStats(eventHandler); |
| } |
| |
| protected abstract Differencer evaluatorDiffer(); |
| |
| @ForOverride |
| protected abstract BuildDriver createBuildDriver(); |
| |
| /** Clear any configured target data stored outside Skyframe. */ |
| public void handleAnalysisInvalidatingChange() { |
| logger.atInfo().log("Dropping configured target data"); |
| analysisCacheDiscarded = true; |
| clearTrimmingCache(); |
| skyframeBuildView.clearInvalidatedActionLookupKeys(); |
| skyframeBuildView.clearLegacyData(); |
| ArtifactNestedSetFunction.getInstance().resetArtifactNestedSetFunctionMaps(); |
| } |
| |
| /** Activates retroactive trimming (idempotently, so has no effect if already active). */ |
| void activateRetroactiveTrimming() { |
| trimmingListener.activate(); |
| } |
| |
| /** Deactivates retroactive trimming (idempotently, so has no effect if already inactive). */ |
| void deactivateRetroactiveTrimming() { |
| trimmingListener.deactivate(); |
| } |
| |
| protected void clearTrimmingCache() { |
| trimmingCache.clear(); |
| } |
| |
| /** Used with dump --rules. */ |
| public static class RuleStat { |
| private final String key; |
| private final String name; |
| private final boolean isRule; |
| private long count; |
| private long actionCount; |
| |
| public RuleStat(String key, String name, boolean isRule) { |
| this.key = key; |
| this.name = name; |
| this.isRule = isRule; |
| } |
| |
| public void addRule(long actionCount) { |
| this.count++; |
| this.actionCount += actionCount; |
| } |
| |
| /** Returns a key that uniquely identifies this rule or aspect. */ |
| public String getKey() { |
| return key; |
| } |
| |
| /** Returns a name for the rule or aspect. */ |
| public String getName() { |
| return name; |
| } |
| |
| /** Returns whether this is a rule or an aspect. */ |
| public boolean isRule() { |
| return isRule; |
| } |
| |
| /** Returns the instance count of this rule or aspect class. */ |
| public long getCount() { |
| return count; |
| } |
| |
| /** Returns the total action count of all instance of this rule or aspect class. */ |
| public long getActionCount() { |
| return actionCount; |
| } |
| } |
| |
| /** Computes statistics on heap-resident rules and aspects. */ |
| public abstract List<RuleStat> getRuleStats(ExtendedEventHandler eventHandler); |
| |
| /** |
| * Decides if graph edges should be stored during this evaluation and checks if the state from the |
| * last evaluation, if any, can be kept. |
| * |
| * <p>If not, it will mark this state for deletion. The actual cleaning is put off until {@link |
| * #sync}, in case no evaluation was actually called for and the existing state can be kept for |
| * longer. |
| */ |
| public void decideKeepIncrementalState( |
| boolean batch, |
| boolean keepStateAfterBuild, |
| boolean trackIncrementalState, |
| boolean discardAnalysisCache, |
| EventHandler eventHandler) { |
| // Assume incrementality. |
| } |
| |
| /** Whether this executor tracks state for the purpose of improving incremental performance. */ |
| public boolean tracksStateForIncrementality() { |
| return true; |
| } |
| |
| /** |
| * Whether this executor may reuse analysis phase nodes for the purpose of improving incremental |
| * performance. |
| * |
| * <p>This may diverge from {@link #tracksStateForIncrementality} in the case where only execution |
| * phase nodes are used for incrementality. |
| */ |
| protected boolean isAnalysisIncremental() { |
| return tracksStateForIncrementality(); |
| } |
| |
| /** |
| * If not null, this is the only source root in the build, corresponding to the single element in |
| * a single-element package path. Such a single-source-root build need not plant the execroot |
| * symlink forest, and can trivially resolve source artifacts from exec paths. As a consequence, |
| * builds where this is not null do not need to track a package -> source root map, and so do not |
| * need to track all loaded packages. |
| */ |
| @Nullable |
| protected Root getForcedSingleSourceRootIfNoExecrootSymlinkCreation() { |
| return null; |
| } |
| |
| private boolean shouldStoreTransitivePackagesInLoadingAndAnalysis() { |
| return getForcedSingleSourceRootIfNoExecrootSymlinkCreation() == null; |
| } |
| |
| @VisibleForTesting |
| protected abstract Injectable injectable(); |
| |
| /** |
| * Types that are created during loading, use significant space, and are definitely not needed |
| * during execution unless explicitly named. |
| * |
| * <p>Some keys, like globs, may be re-evaluated during execution, so these types should only be |
| * discarded if reverse deps are not being tracked! |
| */ |
| private static final ImmutableSet<SkyFunctionName> LOADING_TYPES = |
| ImmutableSet.of( |
| SkyFunctions.PACKAGE, SkyFunctions.BZL_LOAD, SkyFunctions.BZL_COMPILE, SkyFunctions.GLOB); |
| |
| /** Data that should be discarded in {@link #discardPreExecutionCache}. */ |
| protected enum DiscardType { |
| ALL, |
| ANALYSIS_REFS_ONLY, |
| LOADING_NODES_ONLY; |
| |
| boolean discardsAnalysis() { |
| return this != LOADING_NODES_ONLY; |
| } |
| |
| boolean discardsLoading() { |
| return this != ANALYSIS_REFS_ONLY; |
| } |
| } |
| |
| /** |
| * Save memory by removing references to configured targets and aspects in Skyframe. |
| * |
| * <p>These nodes must be recreated on subsequent builds. We do not clear the top-level target |
| * nodes, since their configured targets are needed for the target completion middleman values. |
| * |
| * <p>The nodes are not deleted during this method call, because they are needed for the execution |
| * phase. Instead, their analysis-time data is cleared while preserving the generating action info |
| * needed for execution. The next build will delete the nodes (and recreate them if necessary). |
| * |
| * <p>{@code discardType} can be used to specify which data to discard. |
| */ |
| protected void discardPreExecutionCache( |
| Collection<ConfiguredTarget> topLevelTargets, |
| ImmutableSet<AspectKey> topLevelAspects, |
| DiscardType discardType) { |
| if (discardType.discardsAnalysis()) { |
| topLevelTargets = ImmutableSet.copyOf(topLevelTargets); |
| topLevelAspects = ImmutableSet.copyOf(topLevelAspects); |
| } |
| // This is to prevent throwing away Packages we may need during execution. |
| ImmutableSet.Builder<PackageIdentifier> packageSetBuilder = ImmutableSet.builder(); |
| if (discardType.discardsLoading()) { |
| packageSetBuilder.addAll( |
| Collections2.transform( |
| topLevelTargets, target -> target.getLabel().getPackageIdentifier())); |
| packageSetBuilder.addAll( |
| Collections2.transform( |
| topLevelAspects, aspect -> aspect.getLabel().getPackageIdentifier())); |
| } |
| ImmutableSet<PackageIdentifier> topLevelPackages = packageSetBuilder.build(); |
| try (AutoProfiler p = GoogleAutoProfilerUtils.logged("discarding analysis cache")) { |
| lastAnalysisDiscarded = true; |
| Iterator<? extends Map.Entry<SkyKey, ? extends NodeEntry>> it = |
| memoizingEvaluator.getGraphEntries().iterator(); |
| while (it.hasNext()) { |
| Map.Entry<SkyKey, ? extends NodeEntry> keyAndEntry = it.next(); |
| NodeEntry entry = keyAndEntry.getValue(); |
| if (entry == null || !entry.isDone()) { |
| continue; |
| } |
| SkyKey key = keyAndEntry.getKey(); |
| SkyFunctionName functionName = key.functionName(); |
| if (discardType.discardsLoading()) { |
| // Keep packages for top-level targets and aspects in memory to get the target from later. |
| if (functionName.equals(SkyFunctions.PACKAGE) |
| && topLevelPackages.contains(key.argument())) { |
| continue; |
| } |
| if (LOADING_TYPES.contains(functionName)) { |
| it.remove(); |
| continue; |
| } |
| } |
| if (discardType.discardsAnalysis()) { |
| if (functionName.equals(SkyFunctions.CONFIGURED_TARGET)) { |
| ConfiguredTargetValue ctValue; |
| try { |
| ctValue = (ConfiguredTargetValue) entry.getValue(); |
| } catch (InterruptedException e) { |
| throw new IllegalStateException( |
| "No interruption in in-memory retrieval: " + entry, e); |
| } |
| // ctValue may be null if target was not successfully analyzed. |
| if (ctValue != null) { |
| if (!(ctValue instanceof ActionLookupValue) |
| && discardType.discardsLoading() |
| && !topLevelTargets.contains(ctValue.getConfiguredTarget())) { |
| // If loading is already being deleted, deleting these nodes doesn't hurt. Morally |
| // we should always be able to delete these, since they're not used for execution, |
| // but it leaves the graph inconsistent, and the --discard_analysis_cache with |
| // --track_incremental_state case isn't worth optimizing for. |
| it.remove(); |
| } else { |
| ctValue.clear(!topLevelTargets.contains(ctValue.getConfiguredTarget())); |
| } |
| } |
| } else if (functionName.equals(SkyFunctions.ASPECT)) { |
| AspectValue aspectValue; |
| try { |
| aspectValue = (AspectValue) entry.getValue(); |
| } catch (InterruptedException e) { |
| throw new IllegalStateException( |
| "No interruption in in-memory retrieval: " + entry, e); |
| } |
| // value may be null if target was not successfully analyzed. |
| if (aspectValue != null) { |
| aspectValue.clear(!topLevelAspects.contains(key)); |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| /** |
| * Saves memory by clearing analysis objects from Skyframe. Clears their data without deleting |
| * them (they will be deleted on the next build). May also delete loading-phase objects from the |
| * graph. |
| */ |
| public abstract void clearAnalysisCache( |
| Collection<ConfiguredTarget> topLevelTargets, ImmutableSet<AspectKey> topLevelAspects); |
| |
| protected abstract void dropConfiguredTargetsNow(final ExtendedEventHandler eventHandler); |
| |
| private WorkspaceStatusAction makeWorkspaceStatusAction(String workspaceName) { |
| WorkspaceStatusAction.Environment env = |
| new WorkspaceStatusAction.Environment() { |
| @Override |
| public Artifact createStableArtifact(String name) { |
| ArtifactRoot root = directories.getBuildDataDirectory(workspaceName); |
| return skyframeBuildView |
| .getArtifactFactory() |
| .getDerivedArtifact( |
| PathFragment.create(name), root, WorkspaceStatusValue.BUILD_INFO_KEY); |
| } |
| |
| @Override |
| public Artifact createVolatileArtifact(String name) { |
| ArtifactRoot root = directories.getBuildDataDirectory(workspaceName); |
| return skyframeBuildView |
| .getArtifactFactory() |
| .getConstantMetadataArtifact( |
| PathFragment.create(name), root, WorkspaceStatusValue.BUILD_INFO_KEY); |
| } |
| }; |
| return workspaceStatusActionFactory.createWorkspaceStatusAction(env); |
| } |
| |
| public void injectCoverageReportData(Actions.GeneratingActions actions) { |
| CoverageReportFunction.COVERAGE_REPORT_KEY.set(injectable(), actions.getActions()); |
| } |
| |
| private void setDefaultVisibility(RuleVisibility defaultVisibility) { |
| PrecomputedValue.DEFAULT_VISIBILITY.set(injectable(), defaultVisibility); |
| } |
| |
| private void setConfigSettingVisibilityPolicty(ConfigSettingVisibilityPolicy policy) { |
| PrecomputedValue.CONFIG_SETTING_VISIBILITY_POLICY.set(injectable(), policy); |
| } |
| |
| private void setStarlarkSemantics(StarlarkSemantics starlarkSemantics) { |
| PrecomputedValue.STARLARK_SEMANTICS.set(injectable(), starlarkSemantics); |
| } |
| |
| public void injectExtraPrecomputedValues(List<PrecomputedValue.Injected> extraPrecomputedValues) { |
| for (PrecomputedValue.Injected injected : extraPrecomputedValues) { |
| injected.inject(injectable()); |
| } |
| } |
| |
| protected Cache<PackageIdentifier, LoadedPackageCacheEntry> newPkgFunctionCache() { |
| return CacheBuilder.newBuilder().build(); |
| } |
| |
| protected Cache<PackageIdentifier, PackageFunction.CompiledBuildFile> |
| newCompiledBuildFileCache() { |
| return CacheBuilder.newBuilder().build(); |
| } |
| |
| private void setShowLoadingProgress(boolean showLoadingProgressValue) { |
| showLoadingProgress.set(showLoadingProgressValue); |
| } |
| |
| protected void setCommandId(UUID commandId) { |
| PrecomputedValue.BUILD_ID.set(injectable(), commandId); |
| } |
| |
| /** Returns the build-info.txt and build-changelist.txt artifacts. */ |
| public Collection<Artifact> getWorkspaceStatusArtifacts(ExtendedEventHandler eventHandler) |
| throws InterruptedException { |
| // Should already be present, unless the user didn't request any targets for analysis. |
| EvaluationResult<WorkspaceStatusValue> result = |
| evaluate( |
| ImmutableList.of(WorkspaceStatusValue.BUILD_INFO_KEY), |
| /*keepGoing=*/ true, |
| /*numThreads=*/ 1, |
| eventHandler); |
| WorkspaceStatusValue value = |
| Preconditions.checkNotNull(result.get(WorkspaceStatusValue.BUILD_INFO_KEY)); |
| return ImmutableList.of(value.getStableArtifact(), value.getVolatileArtifact()); |
| } |
| |
| public Map<PathFragment, Root> getArtifactRootsForFiles( |
| final ExtendedEventHandler eventHandler, Iterable<PathFragment> execPaths) |
| throws InterruptedException { |
| return getArtifactRoots(eventHandler, execPaths, true); |
| } |
| |
| public Map<PathFragment, Root> getArtifactRoots( |
| final ExtendedEventHandler eventHandler, Iterable<PathFragment> execPaths) |
| throws InterruptedException { |
| return getArtifactRoots(eventHandler, execPaths, false); |
| } |
| |
| private Map<PathFragment, Root> getArtifactRoots( |
| final ExtendedEventHandler eventHandler, Iterable<PathFragment> execPaths, boolean forFiles) |
| throws InterruptedException { |
| final Map<PathFragment, SkyKey> packageKeys = new HashMap<>(); |
| for (PathFragment execPath : execPaths) { |
| PackageIdentifier pkgIdentifier = |
| PackageIdentifier.discoverFromExecPath(execPath, forFiles, siblingRepositoryLayout); |
| packageKeys.put(execPath, ContainingPackageLookupValue.key(pkgIdentifier)); |
| } |
| |
| EvaluationResult<ContainingPackageLookupValue> result; |
| EvaluationContext evaluationContext = |
| EvaluationContext.newBuilder() |
| .setKeepGoing(true) |
| .setNumThreads(1) |
| .setEventHandler(eventHandler) |
| .build(); |
| |
| synchronized (valueLookupLock) { |
| result = buildDriver.evaluate(packageKeys.values(), evaluationContext); |
| } |
| |
| if (result.hasError()) { |
| return new HashMap<>(); |
| } |
| |
| Map<PathFragment, Root> roots = new HashMap<>(); |
| for (PathFragment execPath : execPaths) { |
| ContainingPackageLookupValue value = result.get(packageKeys.get(execPath)); |
| if (value.hasContainingPackage()) { |
| roots.put(execPath, value.getContainingPackageRoot()); |
| } else { |
| roots.put(execPath, null); |
| } |
| } |
| return roots; |
| } |
| |
| @VisibleForTesting |
| public SkyFunctionEnvironmentForTesting getSkyFunctionEnvironmentForTesting( |
| ExtendedEventHandler eventHandler) { |
| return new SkyFunctionEnvironmentForTesting(eventHandler, this); |
| } |
| |
| public EventBus getEventBus() { |
| return eventBus.get(); |
| } |
| |
| @VisibleForTesting |
| ImmutableList<Root> getPathEntries() { |
| return pkgLocator.get().getPathEntries(); |
| } |
| |
| public AtomicReference<PathPackageLocator> getPackageLocator() { |
| return pkgLocator; |
| } |
| |
| protected abstract void invalidate(Predicate<SkyKey> pred); |
| |
| private static boolean compatibleFileTypes(Dirent.Type oldType, FileStateType newType) { |
| return (oldType.equals(Dirent.Type.FILE) && newType.equals(FileStateType.REGULAR_FILE)) |
| || (oldType.equals(Dirent.Type.UNKNOWN) && newType.equals(FileStateType.SPECIAL_FILE)) |
| || (oldType.equals(Dirent.Type.DIRECTORY) && newType.equals(FileStateType.DIRECTORY)) |
| || (oldType.equals(Dirent.Type.SYMLINK) && newType.equals(FileStateType.SYMLINK)); |
| } |
| |
| protected Differencer.Diff getDiff( |
| TimestampGranularityMonitor tsgm, |
| Collection<PathFragment> modifiedSourceFiles, |
| final Root pathEntry, |
| int fsvcThreads) |
| throws InterruptedException { |
| if (modifiedSourceFiles.isEmpty()) { |
| return new ImmutableDiff(ImmutableList.<SkyKey>of(), ImmutableMap.<SkyKey, SkyValue>of()); |
| } |
| // TODO(bazel-team): change ModifiedFileSet to work with RootedPaths instead of PathFragments. |
| Collection<SkyKey> dirtyFileStateSkyKeys = |
| Collections2.transform( |
| modifiedSourceFiles, |
| pathFragment -> { |
| Preconditions.checkState( |
| !pathFragment.isAbsolute(), "found absolute PathFragment: %s", pathFragment); |
| return FileStateValue.key(RootedPath.toRootedPath(pathEntry, pathFragment)); |
| }); |
| // We only need to invalidate directory values when a file has been created or deleted or |
| // changes type, not when it has merely been modified. Unfortunately we do not have that |
| // information here, so we compute it ourselves. |
| // TODO(bazel-team): Fancy filesystems could provide it with a hypothetically modified |
| // DiffAwareness interface. |
| logger.atInfo().log( |
| "About to recompute filesystem nodes corresponding to files that are known to have " |
| + "changed"); |
| FilesystemValueChecker fsvc = |
| new FilesystemValueChecker(tsgm, /* lastExecutionTimeRange= */ null, fsvcThreads); |
| Map<SkyKey, SkyValue> valuesMap = memoizingEvaluator.getValues(); |
| Differencer.DiffWithDelta diff = |
| fsvc.getNewAndOldValues(valuesMap, dirtyFileStateSkyKeys, new FileDirtinessChecker()); |
| |
| Set<SkyKey> valuesToInvalidate = new HashSet<>(); |
| Map<SkyKey, SkyValue> valuesToInject = new HashMap<>(); |
| for (Map.Entry<SkyKey, Delta> entry : diff.changedKeysWithNewAndOldValues().entrySet()) { |
| SkyKey key = entry.getKey(); |
| Preconditions.checkState(key.functionName().equals(FileStateValue.FILE_STATE), key); |
| RootedPath rootedPath = (RootedPath) key.argument(); |
| Delta delta = entry.getValue(); |
| FileStateValue oldValue = (FileStateValue) delta.getOldValue(); |
| FileStateValue newValue = (FileStateValue) delta.getNewValue(); |
| if (newValue != null) { |
| valuesToInject.put(key, newValue); |
| } else { |
| valuesToInvalidate.add(key); |
| } |
| SkyKey dirListingStateKey = parentDirectoryListingStateKey(rootedPath); |
| // Invalidate the directory listing for the path's parent directory if the change was |
| // relevant (e.g. path turned from a symlink into a directory) OR if we don't have enough |
| // information to determine it was irrelevant. |
| boolean changedType = false; |
| if (newValue == null) { |
| changedType = true; |
| } else if (oldValue != null) { |
| changedType = !oldValue.getType().equals(newValue.getType()); |
| } else { |
| DirectoryListingStateValue oldDirListingStateValue = |
| (DirectoryListingStateValue) valuesMap.get(dirListingStateKey); |
| if (oldDirListingStateValue != null) { |
| String baseName = rootedPath.getRootRelativePath().getBaseName(); |
| Dirent oldDirent = oldDirListingStateValue.getDirents().maybeGetDirent(baseName); |
| changedType = |
| (oldDirent == null) || !compatibleFileTypes(oldDirent.getType(), newValue.getType()); |
| } else { |
| changedType = true; |
| } |
| } |
| if (changedType) { |
| valuesToInvalidate.add(dirListingStateKey); |
| } |
| } |
| for (SkyKey key : diff.changedKeysWithoutNewValues()) { |
| Preconditions.checkState(key.functionName().equals(FileStateValue.FILE_STATE), key); |
| RootedPath rootedPath = (RootedPath) key.argument(); |
| valuesToInvalidate.add(parentDirectoryListingStateKey(rootedPath)); |
| } |
| return new ImmutableDiff(valuesToInvalidate, valuesToInject); |
| } |
| |
| private static SkyKey parentDirectoryListingStateKey(RootedPath rootedPath) { |
| RootedPath parentDirRootedPath = rootedPath.getParentDirectory(); |
| return DirectoryListingStateValue.key(parentDirRootedPath); |
| } |
| |
| /** |
| * Deletes all loaded packages and their upwards transitive closure, forcing reevaluation of all |
| * affected nodes. |
| */ |
| public void clearLoadedPackages() { |
| memoizingEvaluator.delete(k -> SkyFunctions.PACKAGE.equals(k.functionName())); |
| } |
| |
| /** Sets the packages that should be treated as deleted and ignored. */ |
| @VisibleForTesting // productionVisibility = Visibility.PRIVATE |
| public abstract void setDeletedPackages(Iterable<PackageIdentifier> pkgs); |
| |
| /** |
| * Prepares the evaluator for loading. |
| * |
| * <p>MUST be run before every incremental build. |
| */ |
| @VisibleForTesting // productionVisibility = Visibility.PRIVATE |
| public void preparePackageLoading( |
| PathPackageLocator pkgLocator, |
| PackageOptions packageOptions, |
| BuildLanguageOptions buildLanguageOptions, |
| UUID commandId, |
| Map<String, String> clientEnv, |
| TimestampGranularityMonitor tsgm) { |
| Preconditions.checkNotNull(pkgLocator); |
| Preconditions.checkNotNull(tsgm); |
| setActive(true); |
| |
| this.tsgm.set(tsgm); |
| setCommandId(commandId); |
| this.clientEnv.set(clientEnv); |
| |
| setShowLoadingProgress(packageOptions.showLoadingProgress); |
| setDefaultVisibility(packageOptions.defaultVisibility); |
| if (!packageOptions.enforceConfigSettingVisibility) { |
| setConfigSettingVisibilityPolicty(ConfigSettingVisibilityPolicy.LEGACY_OFF); |
| } else { |
| setConfigSettingVisibilityPolicty(packageOptions.configSettingPrivateDefaultVisibility |
| ? ConfigSettingVisibilityPolicy.DEFAULT_STANDARD |
| : ConfigSettingVisibilityPolicy.DEFAULT_PUBLIC); |
| } |
| |
| StarlarkSemantics starlarkSemantics = getEffectiveStarlarkSemantics(buildLanguageOptions); |
| setStarlarkSemantics(starlarkSemantics); |
| setSiblingDirectoryLayout( |
| starlarkSemantics.getBool(BuildLanguageOptions.EXPERIMENTAL_SIBLING_REPOSITORY_LAYOUT)); |
| setPackageLocator(pkgLocator); |
| |
| syscalls.set(getPerBuildSyscallCache(packageOptions.globbingThreads)); |
| this.pkgFactory.setGlobbingThreads(packageOptions.globbingThreads); |
| this.pkgFactory.setMaxDirectoriesToEagerlyVisitInGlobbing( |
| packageOptions.maxDirectoriesToEagerlyVisitInGlobbing); |
| emittedEventState.clear(); |
| |
| // Clear internal caches used by SkyFunctions used for package loading. If the SkyFunctions |
| // never had a chance to restart (e.g. due to user interrupt, or an error in a --nokeep_going |
| // build), these may have stale entries. |
| packageFunctionCache.invalidateAll(); |
| compiledBuildFileCache.invalidateAll(); |
| bzlCompileCache.invalidateAll(); |
| |
| numPackagesLoaded.set(0); |
| if (packageProgress != null) { |
| packageProgress.reset(); |
| } |
| |
| // Reset the stateful SkyframeCycleReporter, which contains cycles from last run. |
| cyclesReporter = createCyclesReporter(); |
| } |
| |
| private void setSiblingDirectoryLayout(boolean experimentalSiblingRepositoryLayout) { |
| this.siblingRepositoryLayout = experimentalSiblingRepositoryLayout; |
| this.artifactFactory.setSiblingRepositoryLayout(experimentalSiblingRepositoryLayout); |
| } |
| |
| public StarlarkSemantics getEffectiveStarlarkSemantics( |
| BuildLanguageOptions buildLanguageOptions) { |
| return buildLanguageOptions.toStarlarkSemantics(); |
| } |
| |
| private void setPackageLocator(PathPackageLocator pkgLocator) { |
| EventBus eventBus = this.eventBus.get(); |
| if (eventBus != null) { |
| eventBus.post(pkgLocator); |
| } |
| |
| PathPackageLocator oldLocator = this.pkgLocator.getAndSet(pkgLocator); |
| PrecomputedValue.PATH_PACKAGE_LOCATOR.set(injectable(), pkgLocator); |
| |
| if (!pkgLocator.equals(oldLocator)) { |
| // The package path is read not only by SkyFunctions but also by some other code paths. |
| // We need to take additional steps to keep the corresponding data structures in sync. |
| // (Some of the additional steps are carried out by ConfiguredTargetValueInvalidationListener, |
| // and some by BuildView#buildHasIncompatiblePackageRoots and #updateSkyframe.) |
| onPkgLocatorChange(oldLocator, pkgLocator); |
| } |
| } |
| |
| protected abstract void onPkgLocatorChange( |
| PathPackageLocator oldLocator, PathPackageLocator pkgLocator); |
| |
| public SkyframeBuildView getSkyframeBuildView() { |
| return skyframeBuildView; |
| } |
| |
| /** Sets the eventBus to use for posting events. */ |
| public void setEventBus(@Nullable EventBus eventBus) { |
| this.eventBus.set(eventBus); |
| } |
| |
| public void setClientEnv(Map<String, String> clientEnv) { |
| this.skyframeActionExecutor.setClientEnv(clientEnv); |
| } |
| |
| /** Sets the path for action log buffers. */ |
| @Deprecated |
| public void setActionOutputRoot(Path actionOutputRoot) { |
| setActionOutputRoot(actionOutputRoot, actionOutputRoot); |
| } |
| |
| /** Sets the path for action log buffers. */ |
| public void setActionOutputRoot(Path actionOutputRoot, Path persistentActionOutputRoot) { |
| Preconditions.checkNotNull(actionOutputRoot); |
| this.actionLogBufferPathGenerator = |
| new ActionLogBufferPathGenerator(actionOutputRoot, persistentActionOutputRoot); |
| this.skyframeActionExecutor.setActionLogBufferPathGenerator(actionLogBufferPathGenerator); |
| } |
| |
| private void setRemoteExecutionEnabled(boolean enabled) { |
| PrecomputedValue.REMOTE_EXECUTION_ENABLED.set(injectable(), enabled); |
| } |
| |
| /** Called when a top-level configuration is determined. */ |
| protected void setTopLevelConfiguration(BuildConfigurationCollection topLevelConfiguration) {} |
| |
| /** |
| * Asks the Skyframe evaluator to build the value for BuildConfigurationCollection and returns the |
| * result. |
| */ |
| // TODO(ulfjack): Remove this legacy method after switching to the Skyframe-based implementation. |
| public BuildConfigurationCollection createConfigurations( |
| ExtendedEventHandler eventHandler, |
| BuildOptions buildOptions, |
| Set<String> multiCpu, |
| boolean keepGoing) |
| throws InvalidConfigurationException { |
| |
| if (configuredTargetProgress != null) { |
| configuredTargetProgress.reset(); |
| } |
| |
| ImmutableList<BuildConfiguration> topLevelTargetConfigs = |
| getConfigurations( |
| eventHandler, |
| PrepareAnalysisPhaseFunction.getTopLevelBuildOptions(buildOptions, multiCpu), |
| buildOptions, |
| keepGoing); |
| |
| BuildConfiguration firstTargetConfig = topLevelTargetConfigs.get(0); |
| |
| BuildOptions targetOptions = firstTargetConfig.getOptions(); |
| BuildOptionsView hostTransitionOptionsView = |
| new BuildOptionsView( |
| firstTargetConfig.getOptions(), HostTransition.INSTANCE.requiresOptionFragments()); |
| BuildOptions hostOptions = |
| targetOptions.get(CoreOptions.class).useDistinctHostConfiguration |
| ? HostTransition.INSTANCE.patch(hostTransitionOptionsView, eventHandler) |
| : targetOptions; |
| BuildConfiguration hostConfig = getConfiguration(eventHandler, hostOptions, keepGoing); |
| |
| // TODO(gregce): cache invalid option errors in BuildConfigurationFunction, then use a dedicated |
| // accessor (i.e. not the event handler) to trigger the exception below. |
| ErrorSensingEventHandler<Void> nosyEventHandler = |
| ErrorSensingEventHandler.withoutPropertyValueTracking(eventHandler); |
| topLevelTargetConfigs.forEach(config -> config.reportInvalidOptions(nosyEventHandler)); |
| if (nosyEventHandler.hasErrors()) { |
| throw new InvalidConfigurationException( |
| "Build options are invalid", Code.INVALID_BUILD_OPTIONS); |
| } |
| return new BuildConfigurationCollection(topLevelTargetConfigs, hostConfig); |
| } |
| |
| /** |
| * Asks the Skyframe evaluator to build the given artifacts and targets, and to test the given |
| * parallel test targets. Additionally, exclusive tests are built together with all the other |
| * tests but they are intentionally *not* run since they must be executed separately one-by-one. |
| */ |
| public EvaluationResult<?> buildArtifacts( |
| Reporter reporter, |
| ResourceManager resourceManager, |
| Executor executor, |
| Set<Artifact> artifactsToBuild, |
| Collection<ConfiguredTarget> targetsToBuild, |
| ImmutableSet<AspectKey> aspects, |
| Set<ConfiguredTarget> parallelTests, |
| Set<ConfiguredTarget> exclusiveTests, |
| OptionsProvider options, |
| ActionCacheChecker actionCacheChecker, |
| TopDownActionCache topDownActionCache, |
| @Nullable EvaluationProgressReceiver executionProgressReceiver, |
| TopLevelArtifactContext topLevelArtifactContext) |
| throws InterruptedException, AbruptExitException { |
| checkActive(); |
| Preconditions.checkState(actionLogBufferPathGenerator != null); |
| |
| deleteActionsIfRemoteOptionsChanged(options); |
| try (SilentCloseable c = |
| Profiler.instance().profile("skyframeActionExecutor.prepareForExecution")) { |
| skyframeActionExecutor.prepareForExecution( |
| reporter, |
| executor, |
| options, |
| actionCacheChecker, |
| topDownActionCache, |
| outputService, |
| isAnalysisIncremental()); |
| } |
| |
| resourceManager.resetResourceUsage(); |
| try { |
| progressReceiver.executionProgressReceiver = executionProgressReceiver; |
| Iterable<TargetCompletionValue.TargetCompletionKey> targetKeys = |
| TargetCompletionValue.keys( |
| targetsToBuild, topLevelArtifactContext, Sets.union(parallelTests, exclusiveTests)); |
| Iterable<SkyKey> aspectKeys = AspectCompletionValue.keys(aspects, topLevelArtifactContext); |
| Iterable<SkyKey> testKeys = |
| TestCompletionValue.keys( |
| parallelTests, topLevelArtifactContext, /*exclusiveTesting=*/ false); |
| EvaluationContext evaluationContext = |
| EvaluationContext.newBuilder() |
| .setKeepGoing(options.getOptions(KeepGoingOption.class).keepGoing) |
| .setNumThreads(options.getOptions(BuildRequestOptions.class).jobs) |
| .setUseForkJoinPool(options.getOptions(BuildRequestOptions.class).useForkJoinPool) |
| .setEventHandler(reporter) |
| .setExecutionPhase() |
| .build(); |
| return buildDriver.evaluate( |
| Iterables.concat(Artifact.keys(artifactsToBuild), targetKeys, aspectKeys, testKeys), |
| evaluationContext); |
| } finally { |
| progressReceiver.executionProgressReceiver = null; |
| // Also releases thread locks. |
| resourceManager.resetResourceUsage(); |
| skyframeActionExecutor.executionOver(); |
| actionExecutionFunction.complete(reporter); |
| } |
| } |
| |
| /** Asks the Skyframe evaluator to run a single exclusive test. */ |
| public EvaluationResult<?> runExclusiveTest( |
| Reporter reporter, |
| ResourceManager resourceManager, |
| Executor executor, |
| ConfiguredTarget exclusiveTest, |
| OptionsProvider options, |
| ActionCacheChecker actionCacheChecker, |
| TopDownActionCache topDownActionCache, |
| @Nullable EvaluationProgressReceiver executionProgressReceiver, |
| TopLevelArtifactContext topLevelArtifactContext) |
| throws InterruptedException { |
| checkActive(); |
| Preconditions.checkState(actionLogBufferPathGenerator != null); |
| |
| try (SilentCloseable c = |
| Profiler.instance().profile("skyframeActionExecutor.prepareForExecution")) { |
| skyframeActionExecutor.prepareForExecution( |
| reporter, |
| executor, |
| options, |
| actionCacheChecker, |
| topDownActionCache, |
| outputService, |
| isAnalysisIncremental()); |
| } |
| |
| resourceManager.resetResourceUsage(); |
| try { |
| Iterable<SkyKey> testKeys = |
| TestCompletionValue.keys( |
| ImmutableSet.of(exclusiveTest), topLevelArtifactContext, /*exclusiveTesting=*/ true); |
| return evaluate( |
| testKeys, |
| /*keepGoing=*/ options.getOptions(KeepGoingOption.class).keepGoing, |
| /*numThreads=*/ options.getOptions(BuildRequestOptions.class).jobs, |
| reporter); |
| } finally { |
| progressReceiver.executionProgressReceiver = null; |
| // Also releases thread locks. |
| resourceManager.resetResourceUsage(); |
| skyframeActionExecutor.executionOver(); |
| actionExecutionFunction.complete(reporter); |
| } |
| } |
| |
| @VisibleForTesting |
| public void prepareBuildingForTestingOnly( |
| Reporter reporter, |
| Executor executor, |
| OptionsProvider options, |
| ActionCacheChecker checker, |
| TopDownActionCache topDownActionCache) { |
| skyframeActionExecutor.prepareForExecution( |
| reporter, |
| executor, |
| options, |
| checker, |
| topDownActionCache, |
| outputService, |
| isAnalysisIncremental()); |
| } |
| |
| private void deleteActionsIfRemoteOptionsChanged(OptionsProvider options) |
| throws AbruptExitException { |
| RemoteOptions remoteOptions = options.getOptions(RemoteOptions.class); |
| Map<String, String> remoteDefaultExecProperties; |
| try { |
| remoteDefaultExecProperties = |
| remoteOptions != null |
| ? remoteOptions.getRemoteDefaultExecProperties() |
| : ImmutableMap.of(); |
| } catch (UserExecException e) { |
| throw new AbruptExitException( |
| DetailedExitCode.of( |
| FailureDetail.newBuilder() |
| .setMessage(e.getMessage()) |
| .setRemoteOptions( |
| FailureDetails.RemoteOptions.newBuilder() |
| .setCode( |
| FailureDetails.RemoteOptions.Code |
| .REMOTE_DEFAULT_EXEC_PROPERTIES_LOGIC_ERROR) |
| .build()) |
| .build()), |
| e); |
| } |
| boolean needsDeletion = |
| lastRemoteDefaultExecProperties != null |
| && !remoteDefaultExecProperties.equals(lastRemoteDefaultExecProperties); |
| lastRemoteDefaultExecProperties = remoteDefaultExecProperties; |
| RemoteOutputsMode remoteOutputsMode = |
| remoteOptions != null ? remoteOptions.remoteOutputsMode : RemoteOutputsMode.ALL; |
| needsDeletion |= lastRemoteOutputsMode != null && lastRemoteOutputsMode != remoteOutputsMode; |
| this.lastRemoteOutputsMode = remoteOutputsMode; |
| if (needsDeletion) { |
| memoizingEvaluator.delete(k -> SkyFunctions.ACTION_EXECUTION.equals(k.functionName())); |
| } |
| } |
| |
| EvaluationResult<SkyValue> targetPatterns( |
| Iterable<? extends SkyKey> patternSkyKeys, |
| int numThreads, |
| boolean keepGoing, |
| ExtendedEventHandler eventHandler) |
| throws InterruptedException { |
| checkActive(); |
| EvaluationContext evaluationContext = |
| EvaluationContext.newBuilder() |
| .setKeepGoing(keepGoing) |
| .setNumThreads(numThreads) |
| .setEventHandler(eventHandler) |
| .setUseForkJoinPool(true) |
| .build(); |
| return buildDriver.evaluate(patternSkyKeys, evaluationContext); |
| } |
| |
| @VisibleForTesting |
| public BuildOptions getDefaultBuildOptions() { |
| return defaultBuildOptions; |
| } |
| |
| /** |
| * Returns the {@link ConfiguredTargetAndData}s corresponding to the given keys. |
| * |
| * <p>For use for legacy support and tests calling through {@code BuildView} only. |
| * |
| * <p>If a requested configured target is in error, the corresponding value is omitted from the |
| * returned list. |
| */ |
| @ThreadSafety.ThreadSafe |
| public ImmutableList<ConfiguredTargetAndData> getConfiguredTargetsForTesting( |
| ExtendedEventHandler eventHandler, |
| BuildConfiguration originalConfig, |
| Iterable<DependencyKey> keys) |
| throws TransitionException, InvalidConfigurationException, InterruptedException { |
| return getConfiguredTargetMapForTesting(eventHandler, originalConfig, keys).values().asList(); |
| } |
| |
| /** |
| * Returns a map from {@link Dependency} inputs to the {@link ConfiguredTargetAndData}s |
| * corresponding to those dependencies. |
| * |
| * <p>For use for legacy support and tests calling through {@code BuildView} only. |
| * |
| * <p>If a requested configured target is in error, the corresponding value is omitted from the |
| * returned list. |
| */ |
| @ThreadSafety.ThreadSafe |
| public ImmutableMultimap<DependencyKey, ConfiguredTargetAndData> getConfiguredTargetMapForTesting( |
| ExtendedEventHandler eventHandler, |
| BuildConfigurationValue.Key originalConfig, |
| Iterable<DependencyKey> keys) |
| throws InvalidConfigurationException, InterruptedException { |
| return getConfiguredTargetMapForTesting( |
| eventHandler, getConfiguration(eventHandler, originalConfig), keys); |
| } |
| |
| /** |
| * Returns a map from {@link Dependency} inputs to the {@link ConfiguredTargetAndData}s |
| * corresponding to those dependencies. |
| * |
| * <p>For use for legacy support and tests calling through {@code BuildView} only. |
| * |
| * <p>If a requested configured target is in error, the corresponding value is omitted from the |
| * returned list except... |
| */ |
| @ThreadSafety.ThreadSafe |
| private ImmutableMultimap<DependencyKey, ConfiguredTargetAndData> |
| getConfiguredTargetMapForTesting( |
| ExtendedEventHandler eventHandler, |
| BuildConfiguration originalConfig, |
| Iterable<DependencyKey> keys) |
| throws InvalidConfigurationException, InterruptedException { |
| checkActive(); |
| |
| Multimap<DependencyKey, BuildConfiguration> configs; |
| if (originalConfig != null) { |
| configs = |
| getConfigurations(eventHandler, originalConfig.getOptions(), keys).getConfigurationMap(); |
| } else { |
| configs = ArrayListMultimap.create(); |
| for (DependencyKey key : keys) { |
| configs.put(key, null); |
| } |
| } |
| |
| final List<SkyKey> skyKeys = new ArrayList<>(); |
| for (DependencyKey key : keys) { |
| if (!configs.containsKey(key)) { |
| // If we couldn't compute a configuration for this target, the target was in error (e.g. |
| // it couldn't be loaded). Exclude it from the results. |
| continue; |
| } |
| for (BuildConfiguration depConfig : configs.get(key)) { |
| skyKeys.add( |
| ConfiguredTargetKey.builder() |
| .setLabel(key.getLabel()) |
| .setConfiguration(depConfig) |
| .build()); |
| for (AspectDeps aspectDeps : key.getAspects().getUsedAspects()) { |
| skyKeys.add( |
| AspectValueKey.createAspectKey( |
| key.getLabel(), depConfig, aspectDeps.getAspect(), depConfig)); |
| } |
| } |
| skyKeys.add(PackageValue.key(key.getLabel().getPackageIdentifier())); |
| } |
| |
| EvaluationResult<SkyValue> result = evaluateSkyKeys(eventHandler, skyKeys); |
| |
| ImmutableMultimap.Builder<DependencyKey, ConfiguredTargetAndData> cts = |
| ImmutableMultimap.builder(); |
| |
| // Logic copied from ConfiguredTargetFunction#computeDependencies. |
| Set<SkyKey> aliasPackagesToFetch = new HashSet<>(); |
| List<DependencyKey> aliasKeysToRedo = new ArrayList<>(); |
| EvaluationResult<SkyValue> aliasPackageValues = null; |
| Iterable<DependencyKey> keysToProcess = keys; |
| for (int i = 0; i < 2; i++) { |
| DependentNodeLoop: |
| for (DependencyKey key : keysToProcess) { |
| if (!configs.containsKey(key)) { |
| // If we couldn't compute a configuration for this target, the target was in error (e.g. |
| // it couldn't be loaded). Exclude it from the results. |
| continue; |
| } |
| for (BuildConfiguration depConfig : configs.get(key)) { |
| SkyKey configuredTargetKey = |
| ConfiguredTargetKey.builder() |
| .setLabel(key.getLabel()) |
| .setConfiguration(depConfig) |
| .build(); |
| if (result.get(configuredTargetKey) == null) { |
| continue; |
| } |
| |
| ConfiguredTarget configuredTarget = |
| ((ConfiguredTargetValue) result.get(configuredTargetKey)).getConfiguredTarget(); |
| Label label = configuredTarget.getLabel(); |
| SkyKey packageKey = PackageValue.key(label.getPackageIdentifier()); |
| PackageValue packageValue; |
| if (i == 0) { |
| packageValue = (PackageValue) result.get(packageKey); |
| if (packageValue == null) { |
| aliasPackagesToFetch.add(packageKey); |
| aliasKeysToRedo.add(key); |
| continue; |
| } |
| } else { |
| packageValue = |
| (PackageValue) |
| Preconditions.checkNotNull(aliasPackageValues.get(packageKey), packageKey); |
| } |
| List<ConfiguredAspect> configuredAspects = new ArrayList<>(); |
| |
| for (AspectDeps aspectDeps : key.getAspects().getUsedAspects()) { |
| SkyKey aspectKey = |
| AspectValueKey.createAspectKey( |
| key.getLabel(), depConfig, aspectDeps.getAspect(), depConfig); |
| if (result.get(aspectKey) == null) { |
| continue DependentNodeLoop; |
| } |
| |
| configuredAspects.add(((AspectValue) result.get(aspectKey)).getConfiguredAspect()); |
| } |
| |
| try { |
| ConfiguredTarget mergedTarget = |
| MergedConfiguredTarget.of(configuredTarget, configuredAspects); |
| BuildConfigurationValue.Key configKey = mergedTarget.getConfigurationKey(); |
| BuildConfiguration resolvedConfig = depConfig; |
| if (configKey == null) { |
| // Unfortunately, it's possible to get a configured target with a null configuration |
| // when depConfig is non-null, so we need to explicitly override it in that case. |
| resolvedConfig = null; |
| } else if (!configKey.equals(BuildConfigurationValue.key(depConfig))) { |
| // Retroactive trimming may change the configuration associated with the dependency. |
| // If it does, we need to get that instance. |
| // TODO(b/140632978): doing these individually instead of doing them all at once may |
| // end up being wasteful use of Skyframe. Although these configurations are guaranteed |
| // to be in the Skyframe cache (because the dependency would have had to retrieve |
| // them to be created in the first place), looking them up repeatedly may be slower |
| // than just keeping a local cache and assigning the same configuration to all the |
| // CTs which need it. Profile this and see if there's a better way. |
| if (!depConfig.trimConfigurationsRetroactively()) { |
| throw new AssertionError( |
| "Loading configurations mid-dependency resolution should ONLY happen when " |
| + "retroactive trimming is enabled."); |
| } |
| resolvedConfig = getConfiguration(eventHandler, mergedTarget.getConfigurationKey()); |
| } |
| cts.put( |
| key, |
| new ConfiguredTargetAndData( |
| mergedTarget, |
| packageValue.getPackage().getTarget(configuredTarget.getLabel().getName()), |
| resolvedConfig, |
| null)); |
| |
| } catch (DuplicateException | NoSuchTargetException e) { |
| throw new IllegalStateException( |
| String.format("Error creating %s", configuredTarget.getLabel()), e); |
| } |
| } |
| } |
| if (aliasKeysToRedo.isEmpty()) { |
| break; |
| } |
| aliasPackageValues = evaluateSkyKeys(eventHandler, aliasPackagesToFetch); |
| keysToProcess = aliasKeysToRedo; |
| } |
| Supplier<Map<BuildConfigurationValue.Key, BuildConfiguration>> configurationLookupSupplier = |
| () -> |
| configs.values().stream() |
| .collect( |
| Collectors.toMap( |
| BuildConfigurationValue::key, java.util.function.Function.identity())); |
| // We ignore the return value here because tests effectively run with --keep_going, and the |
| // loading-phase-error bit is only needed if we're constructing a SkyframeAnalysisResult. |
| SkyframeBuildView.processErrors( |
| result, |
| configurationLookupSupplier, |
| this, |
| eventHandler, |
| /*keepGoing=*/ true, |
| /*eventBus=*/ null); |
| return cts.build(); |
| } |
| |
| /** |
| * Returns the configuration corresponding to the given set of build options. Should not be used |
| * in a world with trimmed configurations. |
| * |
| * @throws InvalidConfigurationException if the build options produces an invalid configuration |
| */ |
| @Deprecated |
| public BuildConfiguration getConfiguration( |
| ExtendedEventHandler eventHandler, BuildOptions options, boolean keepGoing) |
| throws InvalidConfigurationException { |
| return Iterables.getOnlyElement( |
| getConfigurations(eventHandler, ImmutableList.of(options), options, keepGoing)); |
| } |
| |
| public BuildConfiguration getConfiguration( |
| ExtendedEventHandler eventHandler, BuildConfigurationValue.Key configurationKey) { |
| if (configurationKey == null) { |
| return null; |
| } |
| return ((BuildConfigurationValue) |
| evaluateSkyKeys(eventHandler, ImmutableList.of(configurationKey)).get(configurationKey)) |
| .getConfiguration(); |
| } |
| |
| public Map<BuildConfigurationValue.Key, BuildConfiguration> getConfigurations( |
| ExtendedEventHandler eventHandler, Collection<BuildConfigurationValue.Key> keys) { |
| EvaluationResult<SkyValue> evaluationResult = evaluateSkyKeys(eventHandler, keys); |
| return keys.stream() |
| .collect( |
| Collectors.toMap( |
| java.util.function.Function.identity(), |
| (key) -> ((BuildConfigurationValue) evaluationResult.get(key)).getConfiguration())); |
| } |
| /** |
| * Returns the configurations corresponding to the given sets of build options. Output order is |
| * the same as input order. |
| * |
| * @throws InvalidConfigurationException if any build options produces an invalid configuration |
| */ |
| // TODO(ulfjack): Remove this legacy method after switching to the Skyframe-based implementation. |
| private ImmutableList<BuildConfiguration> getConfigurations( |
| ExtendedEventHandler eventHandler, |
| List<BuildOptions> optionsList, |
| BuildOptions referenceBuildOptions, |
| boolean keepGoing) |
| throws InvalidConfigurationException { |
| Preconditions.checkArgument(!Iterables.isEmpty(optionsList)); |
| |
| // Prepare the Skyframe inputs. |
| // TODO(gregce): support trimmed configs. |
| ImmutableSortedSet<Class<? extends Fragment>> allFragments = |
| ruleClassProvider.getConfigurationFragments().stream() |
| .collect( |
| ImmutableSortedSet.toImmutableSortedSet(BuildConfiguration.lexicalFragmentSorter)); |
| |
| PlatformMappingValue platformMappingValue = |
| getPlatformMappingValue(eventHandler, referenceBuildOptions); |
| |
| ImmutableList.Builder<SkyKey> configSkyKeysBuilder = ImmutableList.builder(); |
| for (BuildOptions options : optionsList) { |
| configSkyKeysBuilder.add(toConfigurationKey(platformMappingValue, allFragments, options)); |
| } |
| |
| ImmutableList<SkyKey> configSkyKeys = configSkyKeysBuilder.build(); |
| |
| // Skyframe-evaluate the configurations and throw errors if any. |
| EvaluationResult<SkyValue> evalResult = evaluateSkyKeys(eventHandler, configSkyKeys, keepGoing); |
| if (evalResult.hasError()) { |
| Map.Entry<SkyKey, ErrorInfo> firstError = Iterables.get(evalResult.errorMap().entrySet(), 0); |
| ErrorInfo error = firstError.getValue(); |
| Throwable e = error.getException(); |
| // Wrap loading failed exceptions |
| if (e instanceof NoSuchThingException) { |
| e = new InvalidConfigurationException(((NoSuchThingException) e).getDetailedExitCode(), e); |
| } else if (e == null && !error.getCycleInfo().isEmpty()) { |
| getCyclesReporter().reportCycles(error.getCycleInfo(), firstError.getKey(), eventHandler); |
| e = |
| new InvalidConfigurationException( |
| "cannot load build configuration because of this cycle", Code.CYCLE); |
| } |
| if (e != null) { |
| Throwables.throwIfInstanceOf(e, InvalidConfigurationException.class); |
| } |
| throw new IllegalStateException("Unknown error during configuration creation evaluation", e); |
| } |
| |
| // Prepare and return the results. |
| return configSkyKeys.stream() |
| .map(key -> ((BuildConfigurationValue) evalResult.get(key)).getConfiguration()) |
| .collect(ImmutableList.toImmutableList()); |
| } |
| |
| /** |
| * Retrieves the configurations needed for the given deps. If {@link |
| * BuildConfiguration#trimConfigurations()} is true, trims their fragments to only those needed by |
| * their transitive closures. Else unconditionally includes all fragments. |
| * |
| * <p>Skips targets with loading phase errors. |
| */ |
| // Keep this in sync with {@link PrepareAnalysisPhaseFunction#getConfigurations}. |
| // TODO(ulfjack): Remove this legacy method after switching to the Skyframe-based implementation. |
| @Override |
| public ConfigurationsResult getConfigurations( |
| ExtendedEventHandler eventHandler, BuildOptions fromOptions, Iterable<DependencyKey> keys) |
| throws InvalidConfigurationException, InterruptedException { |
| ConfigurationsResult.Builder builder = ConfigurationsResult.newBuilder(); |
| Set<DependencyKey> depsToEvaluate = new HashSet<>(); |
| |
| ImmutableSortedSet<Class<? extends Fragment>> allFragments = null; |
| if (useUntrimmedConfigs(fromOptions)) { |
| allFragments = ruleClassProvider.getAllFragments(); |
| } |
| |
| // Get the fragments needed for dynamic configuration nodes. |
| final List<SkyKey> transitiveFragmentSkyKeys = new ArrayList<>(); |
| Map<Label, ImmutableSortedSet<Class<? extends Fragment>>> fragmentsMap = new HashMap<>(); |
| Set<Label> labelsWithErrors = new HashSet<>(); |
| for (DependencyKey key : keys) { |
| if (useUntrimmedConfigs(fromOptions)) { |
| fragmentsMap.put(key.getLabel(), allFragments); |
| } else { |
| depsToEvaluate.add(key); |
| transitiveFragmentSkyKeys.add(TransitiveTargetKey.of(key.getLabel())); |
| } |
| } |
| EvaluationResult<SkyValue> fragmentsResult = |
| evaluateSkyKeys(eventHandler, transitiveFragmentSkyKeys, /*keepGoing=*/ true); |
| for (Map.Entry<SkyKey, ErrorInfo> entry : fragmentsResult.errorMap().entrySet()) { |
| reportCycles(eventHandler, entry.getValue().getCycleInfo(), entry.getKey()); |
| } |
| for (DependencyKey key : keys) { |
| if (!depsToEvaluate.contains(key)) { |
| // No fragments to compute here. |
| } else if (fragmentsResult.getError(TransitiveTargetKey.of(key.getLabel())) != null) { |
| labelsWithErrors.add(key.getLabel()); |
| builder.setHasError(); |
| } else { |
| TransitiveTargetValue ttv = |
| (TransitiveTargetValue) fragmentsResult.get(TransitiveTargetKey.of(key.getLabel())); |
| fragmentsMap.put( |
| key.getLabel(), |
| ImmutableSortedSet.copyOf( |
| BuildConfiguration.lexicalFragmentSorter, |
| ttv.getTransitiveConfigFragments().toSet())); |
| } |
| } |
| |
| PlatformMappingValue platformMappingValue = getPlatformMappingValue(eventHandler, fromOptions); |
| |
| // Now get the configurations. |
| final List<SkyKey> configSkyKeys = new ArrayList<>(); |
| for (DependencyKey key : keys) { |
| if (labelsWithErrors.contains(key.getLabel())) { |
| continue; |
| } |
| ImmutableSortedSet<Class<? extends Fragment>> depFragments = fragmentsMap.get(key.getLabel()); |
| if (depFragments != null) { |
| ConfigurationTransition transition = key.getTransition(); |
| if (transition == NullTransition.INSTANCE) { |
| continue; |
| } |
| Collection<BuildOptions> toOptions = Collections.singletonList(fromOptions); |
| try { |
| Map<PackageValue.Key, PackageValue> buildSettingPackages = |
| getBuildSettingPackages(transition, eventHandler); |
| toOptions = |
| ConfigurationResolver.applyTransition( |
| fromOptions, transition, buildSettingPackages, eventHandler) |
| .values(); |
| } catch (TransitionException e) { |
| eventHandler.handle(Event.error(e.getMessage())); |
| builder.setHasError(); |
| continue; |
| } |
| for (BuildOptions toOption : toOptions) { |
| configSkyKeys.add(toConfigurationKey(platformMappingValue, depFragments, toOption)); |
| } |
| } |
| } |
| EvaluationResult<SkyValue> configsResult = |
| evaluateSkyKeys(eventHandler, configSkyKeys, /*keepGoing=*/ true); |
| for (DependencyKey key : keys) { |
| if (labelsWithErrors.contains(key.getLabel())) { |
| continue; |
| } |
| ImmutableSortedSet<Class<? extends Fragment>> depFragments = fragmentsMap.get(key.getLabel()); |
| if (depFragments != null) { |
| if (key.getTransition() == NullTransition.INSTANCE) { |
| builder.put(key, null); |
| continue; |
| } |
| Collection<BuildOptions> toOptions = Collections.singletonList(fromOptions); |
| try { |
| Map<PackageValue.Key, PackageValue> buildSettingPackages = |
| getBuildSettingPackages(key.getTransition(), eventHandler); |
| toOptions = |
| ConfigurationResolver.applyTransition( |
| fromOptions, key.getTransition(), buildSettingPackages, eventHandler) |
| .values(); |
| } catch (TransitionException e) { |
| eventHandler.handle(Event.error(e.getMessage())); |
| builder.setHasError(); |
| continue; |
| } |
| for (BuildOptions toOption : toOptions) { |
| BuildConfigurationValue.Key configKey = |
| toConfigurationKey(platformMappingValue, depFragments, toOption); |
| BuildConfigurationValue configValue = |
| ((BuildConfigurationValue) configsResult.get(configKey)); |
| if (configValue != null) { |
| builder.put(key, configValue.getConfiguration()); |
| } else if (configsResult.errorMap().containsKey(configKey)) { |
| ErrorInfo configError = configsResult.getError(configKey); |
| if (configError.getException() instanceof InvalidConfigurationException) { |
| // Wrap underlying exception to make it clearer to developers which line of code |
| // actually threw exception. |
| InvalidConfigurationException underlying = |
| (InvalidConfigurationException) configError.getException(); |
| throw new InvalidConfigurationException(underlying.getDetailedExitCode(), underlying); |
| } |
| } |
| } |
| } |
| } |
| return builder.build(); |
| } |
| |
| /** Returns every {@link BuildConfigurationValue.Key} in the graph. */ |
| public Collection<SkyKey> getTransitiveConfigurationKeys() { |
| return memoizingEvaluator.getDoneValues().keySet().stream() |
| .filter(key -> SkyFunctions.BUILD_CONFIGURATION.equals(key.functionName())) |
| .collect(ImmutableList.toImmutableList()); |
| } |
| |
| private PlatformMappingValue getPlatformMappingValue( |
| ExtendedEventHandler eventHandler, BuildOptions referenceBuildOptions) |
| throws InvalidConfigurationException { |
| PathFragment platformMappingPath = |
| referenceBuildOptions.get(PlatformOptions.class).platformMappings; |
| |
| PlatformMappingValue.Key platformMappingKey = |
| PlatformMappingValue.Key.create(platformMappingPath); |
| EvaluationResult<SkyValue> evaluationResult = |
| evaluateSkyKeys(eventHandler, ImmutableSet.of(platformMappingKey)); |
| if (evaluationResult.hasError()) { |
| throw new InvalidConfigurationException( |
| Code.PLATFORM_MAPPING_EVALUATION_FAILURE, evaluationResult.getError().getException()); |
| } |
| return (PlatformMappingValue) evaluationResult.get(platformMappingKey); |
| } |
| |
| private BuildConfigurationValue.Key toConfigurationKey( |
| PlatformMappingValue platformMappingValue, |
| ImmutableSortedSet<Class<? extends Fragment>> depFragments, |
| BuildOptions toOption) |
| throws InvalidConfigurationException { |
| try { |
| return BuildConfigurationValue.keyWithPlatformMapping( |
| platformMappingValue, |
| defaultBuildOptions, |
| depFragments, |
| BuildOptions.diffForReconstruction(defaultBuildOptions, toOption)); |
| } catch (OptionsParsingException e) { |
| throw new InvalidConfigurationException(Code.INVALID_BUILD_OPTIONS, e); |
| } |
| } |
| |
| /** Keep in sync with {@link StarlarkTransition#getBuildSettingPackages} */ |
| private Map<PackageValue.Key, PackageValue> getBuildSettingPackages( |
| ConfigurationTransition transition, ExtendedEventHandler eventHandler) |
| throws TransitionException { |
| HashMap<PackageValue.Key, PackageValue> buildSettingPackages = new HashMap<>(); |
| // This happens before cycle detection so keep track of all seen build settings to ensure |
| // we don't get stuck in endless loops (e.g. //alias1->//alias2 && //alias2->alias1) |
| Set<Label> allSeenBuildSettings = new HashSet<>(); |
| ImmutableSet<Label> unverifiedBuildSettings = |
| StarlarkTransition.getAllBuildSettings(transition); |
| while (!unverifiedBuildSettings.isEmpty()) { |
| for (Label buildSetting : unverifiedBuildSettings) { |
| if (!allSeenBuildSettings.add(buildSetting)) { |
| throw new TransitionException( |
| String.format( |
| "Error with aliased build settings related to '%s'. Either your aliases form a" |
| + " dependency cycle or you're attempting to set both an alias its actual" |
| + " target in the same transition.", |
| buildSetting)); |
| } |
| } |
| ImmutableSet<PackageValue.Key> buildSettingPackageKeys = |
| StarlarkTransition.getPackageKeysFromLabels(unverifiedBuildSettings); |
| EvaluationResult<SkyValue> newlyLoaded = |
| evaluateSkyKeys(eventHandler, buildSettingPackageKeys, true); |
| if (newlyLoaded.hasError()) { |
| Map.Entry<SkyKey, ErrorInfo> errorEntry = |
| Preconditions.checkNotNull( |
| Iterables.getFirst(newlyLoaded.errorMap().entrySet(), null), newlyLoaded); |
| throw new TransitionException( |
| new NoSuchPackageException( |
| ((PackageValue.Key) errorEntry.getKey()).argument(), |
| "Unable to find build setting package", |
| errorEntry.getValue().getException())); |
| } |
| buildSettingPackageKeys.forEach( |
| k -> buildSettingPackages.put(k, (PackageValue) newlyLoaded.get(k))); |
| unverifiedBuildSettings = |
| StarlarkTransition.verifyBuildSettingsAndGetAliases( |
| buildSettingPackages, unverifiedBuildSettings); |
| } |
| return buildSettingPackages; |
| } |
| |
| /** |
| * Returns whether configurations should trim their fragments to only those needed by targets and |
| * their transitive dependencies. |
| */ |
| private static boolean useUntrimmedConfigs(BuildOptions options) { |
| return options.get(CoreOptions.class).configsMode == CoreOptions.ConfigsMode.NOTRIM; |
| } |
| |
| /** |
| * Evaluates the given sky keys, blocks, and returns their evaluation results. Fails fast on the |
| * first evaluation error. |
| */ |
| private EvaluationResult<SkyValue> evaluateSkyKeys( |
| final ExtendedEventHandler eventHandler, final Iterable<? extends SkyKey> skyKeys) { |
| return evaluateSkyKeys(eventHandler, skyKeys, false); |
| } |
| |
| /** |
| * Evaluates the given sky keys, blocks, and returns their evaluation results. Enables/disables |
| * "keep going" on evaluation errors as specified. |
| */ |
| EvaluationResult<SkyValue> evaluateSkyKeys( |
| final ExtendedEventHandler eventHandler, |
| final Iterable<? extends SkyKey> skyKeys, |
| final boolean keepGoing) { |
| EvaluationResult<SkyValue> result; |
| try { |
| result = |
| callUninterruptibly( |
| () -> { |
| synchronized (valueLookupLock) { |
| try { |
| skyframeBuildView.enableAnalysis(true); |
| return evaluate( |
| skyKeys, keepGoing, /*numThreads=*/ DEFAULT_THREAD_COUNT, eventHandler); |
| } finally { |
| skyframeBuildView.enableAnalysis(false); |
| } |
| } |
| }); |
| } catch (Exception e) { |
| throw new IllegalStateException(e); // Should never happen. |
| } |
| return result; |
| } |
| |
| /** |
| * Returns a dynamic configuration constructed from the given configuration fragments and build |
| * options. |
| */ |
| @VisibleForTesting |
| public BuildConfiguration getConfigurationForTesting( |
| ExtendedEventHandler eventHandler, FragmentClassSet fragments, BuildOptions options) |
| throws InterruptedException, OptionsParsingException, InvalidConfigurationException { |
| SkyKey key = |
| BuildConfigurationValue.keyWithPlatformMapping( |
| getPlatformMappingValue(eventHandler, options), |
| defaultBuildOptions, |
| fragments, |
| BuildOptions.diffForReconstruction(defaultBuildOptions, options)); |
| BuildConfigurationValue result = |
| (BuildConfigurationValue) |
| evaluate( |
| ImmutableList.of(key), |
| /*keepGoing=*/ false, |
| /*numThreads=*/ DEFAULT_THREAD_COUNT, |
| eventHandler) |
| .get(key); |
| return result.getConfiguration(); |
| } |
| |
| /** Returns a particular configured target. */ |
| @VisibleForTesting |
| @Nullable |
| public ConfiguredTarget getConfiguredTargetForTesting( |
| ExtendedEventHandler eventHandler, Label label, BuildConfiguration configuration) |
| throws TransitionException, InvalidConfigurationException, InterruptedException { |
| return getConfiguredTargetForTesting(eventHandler, label, configuration, NoTransition.INSTANCE); |
| } |
| |
| /** Returns a particular configured target after applying the given transition. */ |
| @VisibleForTesting |
| @Nullable |
| public ConfiguredTarget getConfiguredTargetForTesting( |
| ExtendedEventHandler eventHandler, |
| Label label, |
| BuildConfiguration configuration, |
| ConfigurationTransition transition) |
| throws TransitionException, InvalidConfigurationException, InterruptedException { |
| ConfiguredTargetAndData configuredTargetAndData = |
| getConfiguredTargetAndDataForTesting(eventHandler, label, configuration, transition); |
| return configuredTargetAndData == null ? null : configuredTargetAndData.getConfiguredTarget(); |
| } |
| |
| @VisibleForTesting |
| @Nullable |
| public ConfiguredTargetAndData getConfiguredTargetAndDataForTesting( |
| ExtendedEventHandler eventHandler, |
| Label label, |
| BuildConfiguration configuration, |
| ConfigurationTransition transition) |
| throws TransitionException, InvalidConfigurationException, InterruptedException { |
| |
| ConfigurationTransition transition1 = |
| configuration == null ? NullTransition.INSTANCE : transition; |
| DependencyKey dependencyKey = |
| DependencyKey.builder().setLabel(label).setTransition(transition1).build(); |
| return Iterables.getFirst( |
| getConfiguredTargetsForTesting( |
| eventHandler, configuration, ImmutableList.of(dependencyKey)), |
| null); |
| } |
| |
| @VisibleForTesting |
| @Nullable |
| public ConfiguredTargetAndData getConfiguredTargetAndDataForTesting( |
| ExtendedEventHandler eventHandler, Label label, BuildConfiguration configuration) |
| throws TransitionException, InvalidConfigurationException, InterruptedException { |
| return getConfiguredTargetAndDataForTesting( |
| eventHandler, label, configuration, NoTransition.INSTANCE); |
| } |
| |
| /** |
| * Invalidates Skyframe values corresponding to the given set of modified files under the given |
| * path entry. |
| * |
| * <p>May throw an {@link InterruptedException}, which means that no values have been invalidated. |
| */ |
| @VisibleForTesting |
| public final void invalidateFilesUnderPathForTesting( |
| ExtendedEventHandler eventHandler, ModifiedFileSet modifiedFileSet, Root pathEntry) |
| throws InterruptedException { |
| if (lastAnalysisDiscarded) { |
| // Values were cleared last build, but they couldn't be deleted because they were needed for |
| // the execution phase. We can delete them now. |
| dropConfiguredTargetsNow(eventHandler); |
| lastAnalysisDiscarded = false; |
| } |
| invalidateFilesUnderPathForTestingImpl(eventHandler, modifiedFileSet, pathEntry); |
| } |
| |
| @VisibleForTesting |
| public final void turnOffSyscallCacheForTesting() { |
| syscalls.set(UnixGlob.DEFAULT_SYSCALLS); |
| } |
| |
| protected abstract void invalidateFilesUnderPathForTestingImpl( |
| ExtendedEventHandler eventHandler, ModifiedFileSet modifiedFileSet, Root pathEntry) |
| throws InterruptedException; |
| |
| /** Invalidates SkyFrame values that may have failed for transient reasons. */ |
| public abstract void invalidateTransientErrors(); |
| |
| /** Configures a given set of configured targets. */ |
| EvaluationResult<ActionLookupValue> configureTargets( |
| ExtendedEventHandler eventHandler, |
| List<ConfiguredTargetKey> values, |
| List<AspectValueKey> aspectKeys, |
| boolean keepGoing, |
| int numThreads) |
| throws InterruptedException { |
| checkActive(); |
| |
| eventHandler.post(new ConfigurationPhaseStartedEvent(configuredTargetProgress)); |
| EvaluationContext evaluationContext = |
| EvaluationContext.newBuilder() |
| .setKeepGoing(keepGoing) |
| .setNumThreads(numThreads) |
| .setExecutorServiceSupplier( |
| () -> NamedForkJoinPool.newNamedPool("skyframe-evaluator", numThreads)) |
| .setEventHandler(eventHandler) |
| .build(); |
| EvaluationResult<ActionLookupValue> result = |
| buildDriver.evaluate(Iterables.concat(values, aspectKeys), evaluationContext); |
| // Get rid of any memory retained by the cache -- all loading is done. |
| perBuildSyscallCache.clear(); |
| return result; |
| } |
| |
| /** |
| * Checks the given action lookup values for action conflicts. Values satisfying the returned |
| * predicate are known to be transitively error-free from action conflicts or other analysis |
| * failures. {@link #resetActionConflictsStoredInSkyframe} must be called after this to free |
| * memory coming from this call. |
| */ |
| TopLevelActionConflictReport filterActionConflictsForConfiguredTargetsAndAspects( |
| ExtendedEventHandler eventHandler, |
| Iterable<ActionLookupKey> keys, |
| ImmutableMap<ActionAnalysisMetadata, ArtifactConflictFinder.ConflictException> |
| actionConflicts, |
| TopLevelArtifactContext topLevelArtifactContext) |
| throws InterruptedException { |
| checkActive(); |
| ACTION_CONFLICTS.set(injectable(), actionConflicts); |
| // This work is CPU-bound, so use the number of available processors. |
| EvaluationResult<ActionLookupConflictFindingValue> result = |
| evaluate( |
| TopLevelActionLookupConflictFindingFunction.keys(keys, topLevelArtifactContext), |
| /*keepGoing=*/ true, |
| /*numThreads=*/ ResourceUsage.getAvailableProcessors(), |
| eventHandler); |
| |
| // Remove top-level action-conflict detection values for memory efficiency. Non-top-level ones |
| // are removed below. We are OK with this mini-phase being non-incremental as the failure mode |
| // of action conflict is rare. |
| memoizingEvaluator.delete( |
| SkyFunctionName.functionIs(SkyFunctions.TOP_LEVEL_ACTION_LOOKUP_CONFLICT_FINDING)); |
| return new TopLevelActionConflictReport(result, topLevelArtifactContext); |
| } |
| |
| /** |
| * Encapsulation of the result of #filterActionConflictsForConfiguredTargetsAndAspects() allowing |
| * callers to determine which top-level keys did not have analysis errors and retrieve the |
| * ConflictException for those that keys that specifically have conflicts. |
| */ |
| static final class TopLevelActionConflictReport { |
| |
| private final EvaluationResult<ActionLookupConflictFindingValue> result; |
| private final TopLevelArtifactContext topLevelArtifactContext; |
| |
| TopLevelActionConflictReport( |
| EvaluationResult<ActionLookupConflictFindingValue> result, |
| TopLevelArtifactContext topLevelArtifactContext) { |
| this.result = result; |
| this.topLevelArtifactContext = topLevelArtifactContext; |
| } |
| |
| boolean isErrorFree(ActionLookupKey k) { |
| return result.get( |
| TopLevelActionLookupConflictFindingFunction.Key.create(k, topLevelArtifactContext)) |
| != null; |
| } |
| |
| /** |
| * Get the ConflictException produced for the given ActionLookupKey. Will throw if the given key |
| * {@link #isErrorFree is error-free}. |
| */ |
| Optional<ConflictException> getConflictException(ActionLookupKey k) { |
| ErrorInfo errorInfo = |
| result.getError( |
| TopLevelActionLookupConflictFindingFunction.Key.create(k, topLevelArtifactContext)); |
| Exception e = errorInfo.getException(); |
| return Optional.ofNullable(e instanceof ConflictException ? (ConflictException) e : null); |
| } |
| } |
| |
| /** |
| * Clears all action conflicts stored in skyframe that were discovered by a call to {@link |
| * #filterActionConflictsForConfiguredTargetsAndAspects}. |
| * |
| * <p>This function must be called after a call to {@link |
| * #filterActionConflictsForConfiguredTargetsAndAspects}, either directly (in the case of |
| * no-keep_going evaluations) or indirectly by {@link #filterActionConflictsForTopLevelArtifacts} |
| * in keep_going evaluations. |
| */ |
| public void resetActionConflictsStoredInSkyframe() { |
| memoizingEvaluator.delete( |
| SkyFunctionName.functionIs(SkyFunctions.ACTION_LOOKUP_CONFLICT_FINDING)); |
| } |
| |
| /** |
| * Checks the action lookup values owning the given artifacts for action conflicts. Artifacts |
| * satisfying the returned predicate are known to be transitively free from action conflicts. |
| * {@link #filterActionConflictsForConfiguredTargetsAndAspects} must be called before this is |
| * called in order to populate the known action conflicts. |
| * |
| * <p>This method is only called in keep-going mode, since otherwise any known action conflicts |
| * will immediately fail the build. |
| */ |
| public Predicate<Artifact> filterActionConflictsForTopLevelArtifacts( |
| ExtendedEventHandler eventHandler, Collection<Artifact> artifacts) |
| throws InterruptedException { |
| checkActive(); |
| // This work is CPU-bound, so use the number of available processors. |
| EvaluationResult<ActionLookupConflictFindingValue> result = |
| evaluate( |
| Iterables.transform(artifacts, ActionLookupConflictFindingValue::key), |
| /*keepGoing=*/ true, |
| /*numThreads=*/ ResourceUsage.getAvailableProcessors(), |
| eventHandler); |
| |
| // Remove remaining action-conflict detection values immediately for memory efficiency. |
| resetActionConflictsStoredInSkyframe(); |
| |
| return a -> result.get(ActionLookupConflictFindingValue.key(a)) != null; |
| } |
| |
| /** |
| * For internal use in queries: performs a graph update to make sure the transitive closure of the |
| * specified {@code universeKey} is present in the graph, and returns the {@link |
| * EvaluationResult}. |
| * |
| * <p>The graph update is unconditionally done in keep-going mode, so that the query is guaranteed |
| * a complete graph to work on. |
| */ |
| @Override |
| public EvaluationResult<SkyValue> prepareAndGet( |
| Set<SkyKey> roots, EvaluationContext evaluationContext) throws InterruptedException { |
| EvaluationResult<SkyValue> evaluationResult = |
| buildDriver.evaluate(roots, evaluationContext.getCopyWithKeepGoing(/*keepGoing=*/ true)); |
| return evaluationResult; |
| } |
| |
| /** |
| * Get metadata related to the prepareAndGet() lookup. Resulting data is specific to the |
| * underlying evaluation implementation. |
| */ |
| public String prepareAndGetMetadata( |
| Collection<String> patterns, PathFragment offset, OptionsProvider options) |
| throws AbruptExitException, InterruptedException { |
| return buildDriver.meta(ImmutableList.of(getUniverseKey(patterns, offset)), options); |
| } |
| |
| public Optional<UniverseScope> maybeGetHardcodedUniverseScope() { |
| return Optional.empty(); |
| } |
| |
| @ForOverride |
| protected SkyKey getUniverseKey(Collection<String> patterns, PathFragment offset) { |
| return PrepareDepsOfPatternsValue.key(ImmutableList.copyOf(patterns), offset); |
| } |
| |
| /** Returns the generating action of a given artifact ({@code null} if it's a source artifact). */ |
| private ActionAnalysisMetadata getGeneratingAction( |
| ExtendedEventHandler eventHandler, Artifact artifact) throws InterruptedException { |
| if (artifact.isSourceArtifact()) { |
| return null; |
| } |
| |
| ActionLookupData generatingActionKey = |
| ((Artifact.DerivedArtifact) artifact).getGeneratingActionKey(); |
| |
| synchronized (valueLookupLock) { |
| // Note that this will crash (attempting to run a configured target value builder after |
| // analysis) after a failed --nokeep_going analysis in which the configured target that |
| // failed was a (transitive) dependency of the configured target that should generate |
| // this action. We don't expect callers to query generating actions in such cases. |
| EvaluationResult<ActionLookupValue> result = |
| evaluate( |
| ImmutableList.of(generatingActionKey.getActionLookupKey()), |
| /*keepGoing=*/ false, |
| /*numThreads=*/ ResourceUsage.getAvailableProcessors(), |
| eventHandler); |
| if (result.hasError()) { |
| return null; |
| } |
| ActionLookupValue actionLookupValue = result.get(generatingActionKey.getActionLookupKey()); |
| return actionLookupValue.getActions().get(generatingActionKey.getActionIndex()); |
| } |
| } |
| |
| /** |
| * Returns an action graph. |
| * |
| * <p>For legacy compatibility only. |
| */ |
| public ActionGraph getActionGraph(final ExtendedEventHandler eventHandler) { |
| return new ActionGraph() { |
| @Override |
| public ActionAnalysisMetadata getGeneratingAction(final Artifact artifact) { |
| try { |
| return callUninterruptibly( |
| new Callable<ActionAnalysisMetadata>() { |
| @Override |
| public ActionAnalysisMetadata call() throws InterruptedException { |
| return SkyframeExecutor.this.getGeneratingAction(eventHandler, artifact); |
| } |
| }); |
| } catch (Exception e) { |
| throw new IllegalStateException( |
| "Error getting generating action: " + artifact.prettyPrint(), e); |
| } |
| } |
| }; |
| } |
| |
| public PackageManager getPackageManager() { |
| return packageManager; |
| } |
| |
| @VisibleForTesting |
| public TargetPatternPreloader newTargetPatternPreloader() { |
| return new SkyframeTargetPatternEvaluator(this); |
| } |
| |
| public ActionKeyContext getActionKeyContext() { |
| return actionKeyContext; |
| } |
| |
| // TODO(janakr): Is there a better place for this? |
| public HashFunction getHashFunction() { |
| return fileSystem.getDigestFunction().getHashFunction(); |
| } |
| |
| class SkyframePackageLoader { |
| /** |
| * Looks up a particular package (mostly used after the loading phase, so packages should |
| * already be present, but occasionally used pre-loading phase). Use should be discouraged, |
| * since this cannot be used inside a Skyframe evaluation, and concurrent calls are |
| * synchronized. |
| * |
| * <p>Note that this method needs to be synchronized since InMemoryMemoizingEvaluator.evaluate() |
| * method does not support concurrent calls. |
| */ |
| Package getPackage(ExtendedEventHandler eventHandler, PackageIdentifier pkgName) |
| throws InterruptedException, NoSuchPackageException { |
| synchronized (valueLookupLock) { |
| SkyKey key = PackageValue.key(pkgName); |
| // Any call to this method post-loading phase should either be error-free or be in a |
| // keep_going build, since otherwise the build would have failed during loading. Thus |
| // we set keepGoing=true unconditionally. |
| EvaluationResult<PackageValue> result = |
| evaluate( |
| ImmutableList.of(key), |
| /*keepGoing=*/ true, |
| /*numThreads=*/ DEFAULT_THREAD_COUNT, |
| eventHandler); |
| ErrorInfo error = result.getError(key); |
| if (error != null) { |
| if (!error.getCycleInfo().isEmpty()) { |
| reportCycles(eventHandler, result.getError().getCycleInfo(), key); |
| // This can only happen if a package is freshly loaded outside of the target parsing |
| // or loading phase |
| throw new BuildFileContainsErrorsException( |
| pkgName, "Cycle encountered while loading package " + pkgName); |
| } |
| Throwable e = Preconditions.checkNotNull(error.getException(), "%s %s", pkgName, error); |
| // PackageFunction should be catching, swallowing, and rethrowing all transitive |
| // errors as NoSuchPackageExceptions or constructing packages with errors, since we're in |
| // keep_going mode. |
| Throwables.throwIfInstanceOf(e, NoSuchPackageException.class); |
| throw new IllegalStateException( |
| "Unexpected Exception type from PackageValue for '" |
| + pkgName |
| + "'' with error: " |
| + error, |
| e); |
| } |
| return result.get(key).getPackage(); |
| } |
| } |
| |
| /** Returns whether the given package should be consider deleted and thus should be ignored. */ |
| public boolean isPackageDeleted(PackageIdentifier packageName) { |
| Preconditions.checkState( |
| !packageName.getRepository().isDefault(), "package must be absolute: %s", packageName); |
| return deletedPackages.get().contains(packageName); |
| } |
| } |
| |
| @VisibleForTesting |
| public MemoizingEvaluator getEvaluatorForTesting() { |
| return memoizingEvaluator; |
| } |
| |
| @VisibleForTesting |
| public RuleClassProvider getRuleClassProviderForTesting() { |
| return ruleClassProvider; |
| } |
| |
| @VisibleForTesting |
| public PackageFactory getPackageFactoryForTesting() { |
| return pkgFactory; |
| } |
| |
| @VisibleForTesting |
| public PackageSettings getPackageSettingsForTesting() { |
| return pkgFactory.getPackageSettingsForTesting(); |
| } |
| |
| @VisibleForTesting |
| public BlazeDirectories getBlazeDirectoriesForTesting() { |
| return directories; |
| } |
| |
| @VisibleForTesting |
| ActionExecutionStatusReporter getActionExecutionStatusReporterForTesting() { |
| return statusReporterRef.get(); |
| } |
| |
| /** |
| * Initializes and syncs the graph with the given options, readying it for the next evaluation. |
| * |
| * <p>Returns precomputed information about the workspace if it is available at this stage. This |
| * is an optimization allowing implementations which have such information to make it available |
| * early in the build. |
| */ |
| @Nullable |
| public WorkspaceInfoFromDiff sync( |
| ExtendedEventHandler eventHandler, |
| PackageOptions packageOptions, |
| PathPackageLocator pathPackageLocator, |
| BuildLanguageOptions buildLanguageOptions, |
| UUID commandId, |
| Map<String, String> clientEnv, |
| Map<String, String> repoEnvOption, |
| TimestampGranularityMonitor tsgm, |
| OptionsProvider options) |
| throws InterruptedException, AbruptExitException { |
| getActionEnvFromOptions(options.getOptions(CoreOptions.class)); |
| PrecomputedValue.REPO_ENV.set(injectable(), new LinkedHashMap<>(repoEnvOption)); |
| RemoteOptions remoteOptions = options.getOptions(RemoteOptions.class); |
| setRemoteExecutionEnabled(remoteOptions != null && remoteOptions.isRemoteExecutionEnabled()); |
| syncPackageLoading( |
| packageOptions, |
| pathPackageLocator, |
| buildLanguageOptions, |
| commandId, |
| clientEnv, |
| tsgm, |
| options); |
| |
| if (lastAnalysisDiscarded) { |
| dropConfiguredTargetsNow(eventHandler); |
| lastAnalysisDiscarded = false; |
| } |
| return null; |
| } |
| |
| /** |
| * Updates {@link ArtifactNestedSetFunction} with the value of {@link |
| * BuildRequestOptions#nestedSetAsSkyKeyThreshold}. |
| * |
| * @return whether a change was made |
| */ |
| protected static boolean nestedSetAsSkyKeyOptionsChanged(OptionsProvider options) { |
| BuildRequestOptions buildRequestOptions = options.getOptions(BuildRequestOptions.class); |
| if (buildRequestOptions == null) { |
| return false; |
| } |
| return ArtifactNestedSetFunction.sizeThresholdUpdated( |
| buildRequestOptions.nestedSetAsSkyKeyThreshold); |
| } |
| |
| protected void syncPackageLoading( |
| PackageOptions packageOptions, |
| PathPackageLocator pathPackageLocator, |
| BuildLanguageOptions buildLanguageOptions, |
| UUID commandId, |
| Map<String, String> clientEnv, |
| TimestampGranularityMonitor tsgm, |
| OptionsProvider options) |
| throws AbruptExitException { |
| syncPackageLoadingInternal( |
| packageOptions, pathPackageLocator, buildLanguageOptions, commandId, clientEnv, tsgm); |
| } |
| |
| protected final void syncPackageLoadingInternal( |
| PackageOptions packageOptions, |
| PathPackageLocator pathPackageLocator, |
| BuildLanguageOptions buildLanguageOptions, |
| UUID commandId, |
| Map<String, String> clientEnv, |
| TimestampGranularityMonitor tsgm) { |
| try (SilentCloseable c = Profiler.instance().profile("preparePackageLoading")) { |
| preparePackageLoading( |
| pathPackageLocator, packageOptions, buildLanguageOptions, commandId, clientEnv, tsgm); |
| } |
| try (SilentCloseable c = Profiler.instance().profile("setDeletedPackages")) { |
| setDeletedPackages(packageOptions.getDeletedPackages()); |
| } |
| |
| incrementalBuildMonitor = new SkyframeIncrementalBuildMonitor(); |
| invalidateTransientErrors(); |
| sourceArtifactsSeen.reset(); |
| outputArtifactsSeen.reset(); |
| outputArtifactsFromActionCache.reset(); |
| topLevelArtifactsMetric.reset(); |
| } |
| |
| private void getActionEnvFromOptions(CoreOptions opt) { |
| // ImmutableMap does not support null values, so use a LinkedHashMap instead. |
| LinkedHashMap<String, String> actionEnvironment = new LinkedHashMap<>(); |
| if (opt != null) { |
| for (Map.Entry<String, String> v : opt.actionEnvironment) { |
| actionEnvironment.put(v.getKey(), v.getValue()); |
| } |
| } |
| setActionEnv(actionEnvironment); |
| } |
| |
| @VisibleForTesting |
| public void setActionEnv(Map<String, String> actionEnv) { |
| PrecomputedValue.ACTION_ENV.set(injectable(), actionEnv); |
| } |
| |
| public PathPackageLocator createPackageLocator( |
| ExtendedEventHandler eventHandler, List<String> packagePaths, Path workingDirectory) { |
| return PathPackageLocator.create( |
| directories.getOutputBase(), |
| packagePaths, |
| eventHandler, |
| directories.getWorkspace(), |
| workingDirectory, |
| buildFilesByPriority); |
| } |
| |
| private CyclesReporter createCyclesReporter() { |
| return new CyclesReporter( |
| new TargetCycleReporter(packageManager), |
| new ActionArtifactCycleReporter(packageManager), |
| new TestExpansionCycleReporter(packageManager), |
| new RegisteredToolchainsCycleReporter(), |
| // TODO(ulfjack): The StarlarkModuleCycleReporter swallows previously reported cycles |
| // unconditionally! Is that intentional? |
| new StarlarkModuleCycleReporter()); |
| } |
| |
| CyclesReporter getCyclesReporter() { |
| return cyclesReporter; |
| } |
| |
| /** Convenience method with same semantics as {@link CyclesReporter#reportCycles}. */ |
| public void reportCycles( |
| ExtendedEventHandler eventHandler, Iterable<CycleInfo> cycles, SkyKey topLevelKey) { |
| getCyclesReporter().reportCycles(cycles, topLevelKey, eventHandler); |
| } |
| |
| public void setActionExecutionProgressReportingObjects( |
| @Nullable ProgressSupplier supplier, |
| @Nullable ActionCompletedReceiver completionReceiver, |
| @Nullable ActionExecutionStatusReporter statusReporter) { |
| skyframeActionExecutor.setActionExecutionProgressReportingObjects(supplier, completionReceiver); |
| this.statusReporterRef.set(statusReporter); |
| } |
| |
| public abstract void detectModifiedOutputFiles( |
| ModifiedFileSet modifiedOutputFiles, |
| @Nullable Range<Long> lastExecutionTimeRange, |
| boolean trustRemoteArtifacts, |
| int fsvcThreads) |
| throws AbruptExitException, InterruptedException; |
| |
| /** |
| * Mark dirty values for deletion if they've been dirty for longer than N versions. |
| * |
| * <p>Specifying a value N means, if the current version is V and a value was dirtied (and has |
| * remained so) in version U, and U + N <= V, then the value will be marked for deletion and |
| * purged in version V+1. |
| */ |
| public abstract void deleteOldNodes(long versionWindowForDirtyGc); |
| |
| @Nullable |
| public PackageProgressReceiver getPackageProgressReceiver() { |
| return packageProgress; |
| } |
| |
| /** |
| * Loads the given target patterns without applying any filters (such as removing non-test targets |
| * if {@code --build_tests_only} is set). |
| * |
| * @param eventHandler handler which accepts update events |
| * @param targetPatterns patterns to be loaded |
| * @param threadCount number of threads to use for this skyframe evaluation |
| * @param keepGoing whether to attempt to ignore errors. See also {@link KeepGoingOption} |
| */ |
| public TargetPatternPhaseValue loadTargetPatternsWithoutFilters( |
| ExtendedEventHandler eventHandler, |
| List<String> targetPatterns, |
| PathFragment relativeWorkingDirectory, |
| int threadCount, |
| boolean keepGoing) |
| throws TargetParsingException, InterruptedException { |
| SkyKey key = |
| TargetPatternPhaseValue.keyWithoutFilters( |
| ImmutableList.copyOf(targetPatterns), relativeWorkingDirectory); |
| return getTargetPatternPhaseValue(eventHandler, targetPatterns, threadCount, keepGoing, key); |
| } |
| |
| /** |
| * Loads the given target patterns after applying filters configured through parameters and |
| * options (such as removing non-test targets if {@code --build_tests_only} is set). |
| * |
| * @param eventHandler handler which accepts update events |
| * @param targetPatterns patterns to be loaded |
| * @param threadCount number of threads to use for this skyframe evaluation |
| * @param keepGoing whether to attempt to ignore errors. See also {@link KeepGoingOption} |
| * @param determineTests whether to ignore any targets that aren't tests or test suites |
| */ |
| public TargetPatternPhaseValue loadTargetPatternsWithFilters( |
| ExtendedEventHandler eventHandler, |
| List<String> targetPatterns, |
| PathFragment relativeWorkingDirectory, |
| LoadingOptions options, |
| int threadCount, |
| boolean keepGoing, |
| boolean determineTests) |
| throws TargetParsingException, InterruptedException { |
| SkyKey key = |
| TargetPatternPhaseValue.key( |
| ImmutableList.copyOf(targetPatterns), |
| relativeWorkingDirectory, |
| options.compileOneDependency, |
| options.buildTestsOnly, |
| determineTests, |
| ImmutableList.copyOf(options.buildTagFilterList), |
| options.buildManualTests, |
| options.expandTestSuites, |
| TestFilter.forOptions(options, eventHandler, pkgFactory.getRuleClassNames())); |
| return getTargetPatternPhaseValue(eventHandler, targetPatterns, threadCount, keepGoing, key); |
| } |
| |
| private TargetPatternPhaseValue getTargetPatternPhaseValue( |
| ExtendedEventHandler eventHandler, |
| List<String> targetPatterns, |
| int threadCount, |
| boolean keepGoing, |
| SkyKey key) |
| throws InterruptedException, TargetParsingException { |
| Stopwatch timer = Stopwatch.createStarted(); |
| eventHandler.post(new LoadingPhaseStartedEvent(packageProgress)); |
| EvaluationResult<TargetPatternPhaseValue> evalResult = |
| evaluate(ImmutableList.of(key), keepGoing, threadCount, eventHandler); |
| tryThrowTargetParsingException(eventHandler, targetPatterns, key, evalResult); |
| eventHandler.post(new TargetParsingPhaseTimeEvent(timer.stop().elapsed().toMillis())); |
| return evalResult.get(key); |
| } |
| |
| private void tryThrowTargetParsingException( |
| ExtendedEventHandler eventHandler, |
| List<String> targetPatterns, |
| SkyKey key, |
| EvaluationResult<TargetPatternPhaseValue> evalResult) |
| throws TargetParsingException { |
| if (evalResult.hasError()) { |
| ErrorInfo errorInfo = evalResult.getError(key); |
| TargetParsingException exc; |
| if (!errorInfo.getCycleInfo().isEmpty()) { |
| exc = |
| new TargetParsingException( |
| "cycles detected during target parsing", TargetPatterns.Code.CYCLE); |
| getCyclesReporter().reportCycles(errorInfo.getCycleInfo(), key, eventHandler); |
| // Fallback: we don't know which patterns failed, specifically, so we report the entire |
| // set as being in error. |
| eventHandler.post(PatternExpandingError.failed(targetPatterns, exc.getMessage())); |
| } else { |
| exc = constructNoCycleTargetParsingException(eventHandler, targetPatterns, errorInfo); |
| } |
| throw exc; |
| } |
| } |
| |
| private static TargetParsingException constructNoCycleTargetParsingException( |
| ExtendedEventHandler eventHandler, List<String> targetPatterns, ErrorInfo errorInfo) { |
| Exception e = Preconditions.checkNotNull(errorInfo.getException()); |
| DetailedExitCode detailedExitCode = traverseExceptionChain(e); |
| if (!(e instanceof TargetParsingException)) { |
| // If it's a TargetParsingException, then the TargetPatternPhaseFunction has already |
| // reported the error, so we don't need to report it again. |
| eventHandler.post(PatternExpandingError.failed(targetPatterns, e.getMessage())); |
| } |
| |
| // Following SkyframeTargetPatternEvaluator, we create with a new TargetParsingException either |
| // with an existing DetailedExitCode, or with a FailureDetail Code. |
| Throwable cause = e instanceof TargetParsingException ? e.getCause() : e; |
| return detailedExitCode != null |
| ? new TargetParsingException(e.getMessage(), cause, detailedExitCode) |
| : new TargetParsingException( |
| e.getMessage(), cause, TargetPatterns.Code.TARGET_PATTERN_PARSE_FAILURE); |
| } |
| |
| @Nullable |
| private static DetailedExitCode traverseExceptionChain(Exception topLevelException) { |
| Exception traverseException = topLevelException; |
| DetailedExitCode detailedExitCode = null; |
| int traverseLevel = 0; |
| while (traverseLevel < EXCEPTION_TRAVERSAL_LIMIT) { |
| traverseLevel++; |
| detailedExitCode = DetailedException.getDetailedExitCode(traverseException); |
| if (detailedExitCode != null || traverseException.getCause() == null) { |
| break; |
| } |
| traverseException = (Exception) traverseException.getCause(); |
| } |
| return detailedExitCode; |
| } |
| |
| public PrepareAnalysisPhaseValue prepareAnalysisPhase( |
| ExtendedEventHandler eventHandler, |
| BuildOptions buildOptions, |
| Set<String> multiCpu, |
| Collection<Label> labels) |
| throws InvalidConfigurationException, InterruptedException { |
| FragmentClassSet allFragments = |
| FragmentClassSet.of( |
| ruleClassProvider.getConfigurationFragments().stream() |
| .collect( |
| ImmutableSortedSet.toImmutableSortedSet( |
| BuildConfiguration.lexicalFragmentSorter))); |
| SkyKey key = |
| PrepareAnalysisPhaseValue.key( |
| allFragments, |
| BuildOptions.diffForReconstruction(defaultBuildOptions, buildOptions), |
| multiCpu, |
| labels); |
| EvaluationResult<PrepareAnalysisPhaseValue> evalResult = |
| evaluate( |
| ImmutableList.of(key), |
| /*keepGoing=*/ true, |
| /*numThreads=*/ DEFAULT_THREAD_COUNT, |
| eventHandler); |
| if (evalResult.hasError()) { |
| ErrorInfo errorInfo = evalResult.getError(key); |
| Exception e = errorInfo.getException(); |
| if (e == null && !errorInfo.getCycleInfo().isEmpty()) { |
| getCyclesReporter().reportCycles(errorInfo.getCycleInfo(), key, eventHandler); |
| e = |
| new InvalidConfigurationException( |
| "cannot load build configuration because of this cycle", Code.CYCLE); |
| } else if (e instanceof NoSuchThingException) { |
| e = new InvalidConfigurationException(((NoSuchThingException) e).getDetailedExitCode(), e); |
| } |
| if (e != null) { |
| Throwables.throwIfInstanceOf(e, InvalidConfigurationException.class); |
| } |
| throw new IllegalStateException("Unknown error during configuration creation evaluation", e); |
| } |
| |
| if (configuredTargetProgress != null) { |
| configuredTargetProgress.reset(); |
| } |
| |
| PrepareAnalysisPhaseValue prepareAnalysisPhaseValue = evalResult.get(key); |
| return prepareAnalysisPhaseValue; |
| } |
| |
| /** A progress receiver to track analysis invalidation and update progress messages. */ |
| protected class SkyframeProgressReceiver implements EvaluationProgressReceiver { |
| /** |
| * This flag is needed in order to avoid invalidating legacy data when we clear the analysis |
| * cache because of --discard_analysis_cache flag. For that case we want to keep the legacy data |
| * but get rid of the Skyframe data. |
| */ |
| protected boolean ignoreInvalidations = false; |
| /** This receiver is only needed for execution, so it is null otherwise. */ |
| @Nullable EvaluationProgressReceiver executionProgressReceiver = null; |
| /** This receiver is only needed for loading, so it is null otherwise. */ |
| @Override |
| public void invalidated(SkyKey skyKey, InvalidationState state) { |
| trimmingListener.invalidated(skyKey, state); |
| if (ignoreInvalidations) { |
| return; |
| } |
| skyframeBuildView.getProgressReceiver().invalidated(skyKey, state); |
| } |
| |
| @Override |
| public void enqueueing(SkyKey skyKey) { |
| trimmingListener.enqueueing(skyKey); |
| if (ignoreInvalidations) { |
| return; |
| } |
| skyframeBuildView.getProgressReceiver().enqueueing(skyKey); |
| if (executionProgressReceiver != null) { |
| executionProgressReceiver.enqueueing(skyKey); |
| } |
| } |
| |
| @Override |
| public void evaluated( |
| SkyKey skyKey, |
| @Nullable SkyValue newValue, |
| @Nullable ErrorInfo newError, |
| Supplier<EvaluationSuccessState> evaluationSuccessState, |
| EvaluationState state) { |
| trimmingListener.evaluated(skyKey, newValue, newError, evaluationSuccessState, state); |
| if (ignoreInvalidations) { |
| return; |
| } |
| skyframeBuildView |
| .getProgressReceiver() |
| .evaluated(skyKey, newValue, newError, evaluationSuccessState, state); |
| if (executionProgressReceiver != null) { |
| executionProgressReceiver.evaluated( |
| skyKey, newValue, newError, evaluationSuccessState, state); |
| } |
| } |
| } |
| |
| public final ExecutionFinishedEvent createExecutionFinishedEvent() { |
| return createExecutionFinishedEventInternal() |
| .setSourceArtifactsRead(sourceArtifactsSeen.toFilesMetricAndReset()) |
| .setOutputArtifactsSeen(outputArtifactsSeen.toFilesMetricAndReset()) |
| .setOutputArtifactsFromActionCache(outputArtifactsFromActionCache.toFilesMetricAndReset()) |
| .setTopLevelArtifacts(topLevelArtifactsMetric.toFilesMetricAndReset()) |
| .build(); |
| } |
| |
| @ForOverride |
| protected ExecutionFinishedEvent.Builder createExecutionFinishedEventInternal() { |
| return ExecutionFinishedEvent.builderWithDefaults(); |
| } |
| |
| final AnalysisTraversalResult getActionLookupValuesInBuild( |
| List<ConfiguredTargetKey> topLevelCtKeys, List<AspectValueKey> aspectKeys) |
| throws InterruptedException { |
| AnalysisTraversalResult result = new AnalysisTraversalResult(); |
| if (!isAnalysisIncremental()) { |
| // If we do not have incremental analysis state we do not have graph edges, so we cannot |
| // traverse the graph and find only actions in the current build. In this case we can simply |
| // return all ActionLookupValues in the graph, since the graph's lifetime is a single build |
| // anyway. |
| for (Map.Entry<SkyKey, SkyValue> entry : memoizingEvaluator.getDoneValues().entrySet()) { |
| if ((entry.getKey() instanceof ActionLookupKey) && entry.getValue() != null) { |
| result.accumulate(entry.getValue(), (ActionLookupKey) entry.getKey()); |
| } |
| } |
| return result; |
| } |
| WalkableGraph walkableGraph = SkyframeExecutorWrappingWalkableGraph.of(this); |
| Set<SkyKey> seen = CompactHashSet.create(); |
| for (ConfiguredTargetKey key : topLevelCtKeys) { |
| findActionsRecursively(walkableGraph, key, seen, result); |
| } |
| for (AspectValueKey key : aspectKeys) { |
| findActionsRecursively(walkableGraph, key, seen, result); |
| } |
| return result; |
| } |
| |
| static class AnalysisTraversalResult { |
| // Some metrics indicate this is a rough average # of ALVs in a build. |
| private final Sharder<ActionLookupValue> actionShards = new Sharder<>(NUM_JOBS, 200_000); |
| |
| // Metrics. |
| private int configuredObjectCount = 0; |
| private int configuredTargetCount = 0; |
| private int actionCount = 0; |
| private int actionCountNotIncludingAspects = 0; |
| private int inputFileConfiguredTargetCount = 0; |
| private int outputFileConfiguredTargetCount = 0; |
| private int otherConfiguredTargetCount = 0; |
| |
| private AnalysisTraversalResult() {} |
| |
| private void accumulate(SkyValue value, ActionLookupKey keyForDebugging) { |
| boolean isConfiguredTarget = value instanceof ConfiguredTargetValue; |
| boolean isActionLookupValue = value instanceof ActionLookupValue; |
| if (!isConfiguredTarget && !isActionLookupValue) { |
| BugReport.sendBugReport( |
| new IllegalStateException( |
| String.format( |
| "Should only be called with ConfiguredTargetValue or ActionLookupValue: %s %s" |
| + " %s", |
| value.getClass(), keyForDebugging, value))); |
| return; |
| } |
| configuredObjectCount++; |
| if (isConfiguredTarget) { |
| configuredTargetCount++; |
| } |
| if (isActionLookupValue) { |
| ActionLookupValue alv = (ActionLookupValue) value; |
| int numActions = alv.getNumActions(); |
| actionCount += numActions; |
| if (isConfiguredTarget) { |
| actionCountNotIncludingAspects += numActions; |
| } |
| actionShards.add(alv); |
| return; |
| } |
| if (!(value instanceof NonRuleConfiguredTargetValue)) { |
| BugReport.sendBugReport( |
| new IllegalStateException( |
| String.format( |
| "Unexpected value type: %s %s %s", value.getClass(), keyForDebugging, value))); |
| return; |
| } |
| ConfiguredTarget configuredTarget = |
| ((NonRuleConfiguredTargetValue) value).getConfiguredTarget(); |
| if (configuredTarget instanceof InputFileConfiguredTarget) { |
| inputFileConfiguredTargetCount++; |
| } else if (configuredTarget instanceof OutputFileConfiguredTarget) { |
| outputFileConfiguredTargetCount++; |
| } else { |
| otherConfiguredTargetCount++; |
| } |
| } |
| |
| Sharder<ActionLookupValue> getActionShards() { |
| return actionShards; |
| } |
| |
| BuildEventStreamProtos.BuildMetrics.BuildGraphMetrics.Builder getMetrics() { |
| return BuildEventStreamProtos.BuildMetrics.BuildGraphMetrics.newBuilder() |
| .setActionLookupValueCount(configuredObjectCount) |
| .setActionLookupValueCountNotIncludingAspects(configuredTargetCount) |
| .setActionCount(actionCount) |
| .setActionCountNotIncludingAspects(actionCountNotIncludingAspects) |
| .setInputFileConfiguredTargetCount(inputFileConfiguredTargetCount) |
| .setOutputFileConfiguredTargetCount(outputFileConfiguredTargetCount) |
| .setOtherConfiguredTargetCount(otherConfiguredTargetCount); |
| } |
| } |
| |
| private static void findActionsRecursively( |
| WalkableGraph walkableGraph, SkyKey key, Set<SkyKey> seen, AnalysisTraversalResult result) |
| throws InterruptedException { |
| if (!(key instanceof ActionLookupKey) || !seen.add(key)) { |
| // The subgraph of dependencies of ActionLookupKeys never has a non-ActionLookupKey depending |
| // on an ActionLookupKey. So we can skip any non-ActionLookupKeys in the traversal as an |
| // optimization. |
| return; |
| } |
| SkyValue value = walkableGraph.getValue(key); |
| if (value == null) { |
| return; // The value failed to evaluate. |
| } |
| result.accumulate(value, (ActionLookupKey) key); |
| for (SkyKey dep : walkableGraph.getDirectDeps(key)) { |
| findActionsRecursively(walkableGraph, dep, seen, result); |
| } |
| } |
| |
| private <T extends SkyValue> EvaluationResult<T> evaluate( |
| Iterable<? extends SkyKey> roots, |
| boolean keepGoing, |
| int numThreads, |
| ExtendedEventHandler eventHandler) |
| throws InterruptedException { |
| EvaluationContext evaluationContext = |
| EvaluationContext.newBuilder() |
| .setKeepGoing(keepGoing) |
| .setNumThreads(numThreads) |
| .setEventHandler(eventHandler) |
| .build(); |
| return buildDriver.evaluate(roots, evaluationContext); |
| } |
| } |