1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
// Copyright 2015 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.analysis;
import com.google.common.collect.HashMultimap;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Multimap;
import com.google.devtools.build.lib.actions.Action;
import com.google.devtools.build.lib.actions.Artifact;
import com.google.devtools.build.lib.analysis.ExtraActionArtifactsProvider.ExtraArtifactSet;
import com.google.devtools.build.lib.analysis.RuleConfiguredTarget.Mode;
import com.google.devtools.build.lib.analysis.config.BuildConfiguration;
import com.google.devtools.build.lib.cmdline.Label;
import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder;
import com.google.devtools.build.lib.rules.extra.ExtraActionMapProvider;
import com.google.devtools.build.lib.rules.extra.ExtraActionSpec;
import java.util.List;
/**
* A collection of static methods related to ExtraActions.
*/
class ExtraActionUtils {
/**
* Scans {@code action_listeners} associated with this build to see if any
* {@code extra_actions} should be added to this configured target. If any
* action_listeners are present, a partial visit of the artifact/action graph
* is performed (for as long as actions found are owned by this {@link
* ConfiguredTarget}). Any actions that match the {@code action_listener}
* get an {@code extra_action} associated. The output artifacts of the
* extra_action are reported to the {@link AnalysisEnvironment} for
* bookkeeping.
*/
static ExtraActionArtifactsProvider createExtraActionProvider(RuleContext ruleContext) {
BuildConfiguration configuration = ruleContext.getConfiguration();
if (configuration.isHostConfiguration()) {
return ExtraActionArtifactsProvider.EMPTY;
}
ImmutableList<Artifact> extraActionArtifacts = ImmutableList.of();
NestedSetBuilder<ExtraArtifactSet> builder = NestedSetBuilder.stableOrder();
List<Label> actionListenerLabels = configuration.getActionListeners();
if (!actionListenerLabels.isEmpty()
&& ruleContext.attributes().getAttributeDefinition(":action_listener") != null) {
ExtraActionsVisitor visitor =
new ExtraActionsVisitor(ruleContext, computeMnemonicsToExtraActionMap(ruleContext));
// The action list is modified within the body of the loop by the addExtraAction() call,
// thus the copy
for (Action action : ImmutableList.copyOf(
ruleContext.getAnalysisEnvironment().getRegisteredActions())) {
visitor.addExtraAction(action);
}
extraActionArtifacts = visitor.getAndResetExtraArtifacts();
if (!extraActionArtifacts.isEmpty()) {
builder.add(ExtraArtifactSet.of(ruleContext.getLabel(), extraActionArtifacts));
}
}
// Add extra action artifacts from dependencies
for (ExtraActionArtifactsProvider provider : AnalysisUtils.getProviders(
ruleContext.getConfiguredTargetMap().values(), ExtraActionArtifactsProvider.class)) {
builder.addTransitive(provider.getTransitiveExtraActionArtifacts());
}
return ExtraActionArtifactsProvider.create(
NestedSetBuilder.<Artifact>stableOrder().addAll(extraActionArtifacts).build(),
builder.build());
}
/**
* Populates the configuration specific mnemonicToExtraActionMap
* based on all action_listers selected by the user (via the blaze option
* {@code --experimental_action_listener=<target>}).
*/
private static Multimap<String, ExtraActionSpec> computeMnemonicsToExtraActionMap(
RuleContext ruleContext) {
// We copy the multimap here every time. This could be expensive.
Multimap<String, ExtraActionSpec> mnemonicToExtraActionMap = HashMultimap.create();
for (TransitiveInfoCollection actionListener :
ruleContext.getPrerequisites(":action_listener", Mode.HOST)) {
ExtraActionMapProvider provider = actionListener.getProvider(ExtraActionMapProvider.class);
if (provider == null) {
ruleContext.ruleError(String.format(
"Unable to match experimental_action_listeners to this rule. "
+ "Specified target %s is not an action_listener rule",
actionListener.getLabel().toString()));
} else {
mnemonicToExtraActionMap.putAll(provider.getExtraActionMap());
}
}
return mnemonicToExtraActionMap;
}
}
|
