Allow to filter attributes based on the rule of the attribute.

--
MOS_MIGRATED_REVID=111854804

1 files changed, 2 insertions, 2 deletions

diff --git a/src/main/java/com/google/devtools/build/lib/query2/output/ProtoOutputFormatter.java b/src/main/java/com/google/devtools/build/lib/query2/output/ProtoOutputFormatter.java
index e1f8ea78a5..da5a2fab6a 100644
--- a/src/main/java/com/google/devtools/build/lib/query2/output/ProtoOutputFormatter.java
+++ b/src/main/java/com/google/devtools/build/lib/query2/output/ProtoOutputFormatter.java
@@ -139,7 +139,7 @@ public class ProtoOutputFormatter extends AbstractUnorderedFormatter {
           .setLocation(location);
       for (Attribute attr : rule.getAttributes()) {
         if (!includeDefaultValues && !rule.isAttributeValueExplicitlySpecified(attr)
-            || !includeAttribute(attr)) {
+            || !includeAttribute(rule, attr)) {
           continue;
         }
         Iterable<Object> possibleAttributeValues =
@@ -325,7 +325,7 @@ public class ProtoOutputFormatter extends AbstractUnorderedFormatter {
   protected void postProcess(Rule rule, Build.Rule.Builder rulePb) { }
 
   /** Filter out some attributes */
-  protected boolean includeAttribute(Attribute attr) {
+  protected boolean includeAttribute(Rule rule, Attribute attr) {
     return true;
   }
 }