commit e804017809ed3990bbb2f5e6b6cffdcf037349f6
author Dmitry Lomov <dslomov@google.com> Wed Apr 06 14:53:43 2016 +0000
committer Lukacs Berki <lberki@google.com> Thu Apr 07 11:46:14 2016 +0000
tree 539c968a7b80d74cd4799905d6464b8de55a1691
parent b5ce0bb100a9527da156fafa9e0813b3dfe00a2b

Remove AspectClass.getDefinition

Aspect becomes a triple (AspectClass, AspectDefinition,
AspectParameters) and loses its equals() method.
After this CL, SkylarkAspectClass.getDefintion still exists and is
deprecated.

--
MOS_MIGRATED_REVID=119159653

src/main/java/com/google/devtools/build/lib/packages/Attribute.java

diff --git a/src/main/java/com/google/devtools/build/lib/packages/Attribute.java b/src/main/java/com/google/devtools/build/lib/packages/Attribute.java
index ffe0369..bf4a0a7 100644
--- a/src/main/java/com/google/devtools/build/lib/packages/Attribute.java
+++ b/src/main/java/com/google/devtools/build/lib/packages/Attribute.java
@@ -61,21 +61,41 @@
 
   public static final Predicate<RuleClass> NO_RULE = Predicates.alwaysFalse();
 
-  private static final class RuleAspect {
-    private final AspectClass aspectFactory;
-    private final Function<Rule, AspectParameters> parametersExtractor;
+  private abstract static class RuleAspect<C extends AspectClass> {
+    protected final C aspectClass;
+    protected final Function<Rule, AspectParameters> parametersExtractor;
 
-    RuleAspect(AspectClass aspectFactory, Function<Rule, AspectParameters> parametersExtractor) {
-      this.aspectFactory = aspectFactory;
+    protected RuleAspect(C aspectClass, Function<Rule, AspectParameters> parametersExtractor) {
+      this.aspectClass = aspectClass;
       this.parametersExtractor = parametersExtractor;
     }
 
-    AspectClass getAspectFactory() {
-      return aspectFactory;
+    public abstract Aspect getAspect(Rule rule);
+  }
+
+  private static class NativeRuleAspect extends RuleAspect<NativeAspectClass<?>> {
+    public NativeRuleAspect(NativeAspectClass<?> aspectClass,
+        Function<Rule, AspectParameters> parametersExtractor) {
+      super(aspectClass, parametersExtractor);
     }
-    
-    Function<Rule, AspectParameters> getParametersExtractor() {
-      return parametersExtractor;
+
+    @Override
+    public Aspect getAspect(Rule rule) {
+      return Aspect.forNative(aspectClass, parametersExtractor.apply(rule));
+    }
+  }
+
+  private static class SkylarkRuleAspect extends RuleAspect<SkylarkAspectClass> {
+    public SkylarkRuleAspect(SkylarkAspectClass aspectClass) {
+      super(aspectClass, NO_PARAMETERS);
+    }
+
+    @Override
+    public Aspect getAspect(Rule rule) {
+      return Aspect.forSkylark(
+          aspectClass,
+          aspectClass.getDefinition(),
+          parametersExtractor.apply(rule));
     }
   }
 
@@ -281,6 +301,15 @@
     }
   }
 
+  private static final Function<Rule, AspectParameters> NO_PARAMETERS =
+      new Function<Rule, AspectParameters>() {
+        @Override
+        public AspectParameters apply(Rule input) {
+          return AspectParameters.EMPTY;
+        }
+      };
+
+
   /**
    * Creates a new attribute builder.
    *
@@ -301,6 +330,7 @@
    * already undocumented based on its name cannot be marked as undocumented.
    */
   public static class Builder <TYPE> {
+
     private String name;
     private final Type<TYPE> type;
     private Transition configTransition = ConfigurationTransition.NONE;
@@ -724,13 +754,7 @@
      * through this attribute.
      */
     public <T extends NativeAspectFactory> Builder<TYPE> aspect(Class<T> aspect) {
-      Function<Rule, AspectParameters> noParameters = new Function<Rule, AspectParameters>() {
-        @Override
-        public AspectParameters apply(Rule input) {
-          return AspectParameters.EMPTY;
-        }
-      };
-      return this.aspect(aspect, noParameters);
+      return this.aspect(aspect, NO_PARAMETERS);
     }
 
     /**
@@ -750,8 +774,9 @@
      *
      * @param evaluator function that extracts aspect parameters from rule.
      */
-    public Builder<TYPE> aspect(AspectClass aspect, Function<Rule, AspectParameters> evaluator) {
-      this.aspects.add(new RuleAspect(aspect, evaluator));
+    public Builder<TYPE> aspect(
+        NativeAspectClass<?> aspect, Function<Rule, AspectParameters> evaluator) {
+      this.aspects.add(new NativeRuleAspect(aspect, evaluator));
       return this;
     }
 
@@ -759,7 +784,7 @@
      * Asserts that a particular parameterized aspect probably needs to be computed for all direct
      * dependencies through this attribute.
      */
-    public Builder<TYPE> aspect(AspectClass aspect) {
+    public Builder<TYPE> aspect(NativeAspectClass<?> aspect) {
       Function<Rule, AspectParameters> noParameters =
           new Function<Rule, AspectParameters>() {
             @Override
@@ -770,6 +795,11 @@
       return this.aspect(aspect, noParameters);
     }
 
+    public Builder<TYPE> aspect(SkylarkAspectClass aspectClass) {
+      this.aspects.add(new SkylarkRuleAspect(aspectClass));
+      return this;
+    }
+
     /**
      * Sets the predicate-like edge validity checker.
      */
@@ -1400,8 +1430,7 @@
   public ImmutableList<Aspect> getAspects(Rule rule) {
     ImmutableList.Builder<Aspect> builder = ImmutableList.builder();
     for (RuleAspect aspect : aspects) {
-      builder.add(
-          new Aspect(aspect.getAspectFactory(), aspect.getParametersExtractor().apply(rule)));
+      builder.add(aspect.getAspect(rule));
     }
     return builder.build();
   }